From 054428243ccb1c0b39c7fdf8faa449ce415789ce Mon Sep 17 00:00:00 2001 From: Nguyễn Gia Phong Date: Sat, 1 Dec 2018 02:32:41 +0700 Subject: Chemistry is fun --- toys/SPoNGeMoCK.py | 60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100755 toys/SPoNGeMoCK.py (limited to 'toys') diff --git a/toys/SPoNGeMoCK.py b/toys/SPoNGeMoCK.py new file mode 100755 index 0000000..23c943d --- /dev/null +++ b/toys/SPoNGeMoCK.py @@ -0,0 +1,60 @@ +#!/usr/bin/env python3 +"""SPoNGeMoCK stdin using symbols of chemical elements +and print the result to stdout. +""" + +import re +from bisect import bisect_left as bisect +from sys import stdin, stdout + +ELEMENTS = ( + 'H', 'He', 'Li', 'Be', 'B', 'C', 'N', 'O', 'F', 'Ne', 'Na', 'Mg', 'Al', + 'Si', 'P', 'S', 'Cl', 'Ar', 'K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', 'Mn', 'Fe', + 'Co', 'Ni', 'Cu', 'Zn', 'Ga', 'Ge', 'As', 'Se', 'Br', 'Kr', 'Rb', 'Sr', + 'Y', 'Zr', 'Nb', 'Mo', 'Tc', 'Ru', 'Rh', 'Pd' , 'Ag', 'Cd', 'In', 'Sn', + 'Sb', 'Te', 'I', 'Xe', 'Cs', 'Ba', 'La', 'Ce', 'Pr', 'Nd', 'Pm ', 'Sm', + 'Eu', 'Gd', 'Tb', 'Dy', 'Ho', 'Er', 'Tm', 'Yb', 'Lu', 'Hf', 'Ta', 'W', + 'Re', 'Os', 'Ir', 'Pt', 'Au', 'Hg', 'Tl', 'Pb', 'Bi', 'Po', 'At', 'Rn', + 'Fr', 'Ra', 'Ac', 'Th', 'Pa', 'U', 'Np', 'Pu', 'Am', 'Cm', 'Bk', 'Cf', + 'Es', 'Fm', 'Md', 'No', 'Lr', 'Rf', 'Db', 'Sg', 'Bh', 'Hs', 'Mt', 'Ds', + 'Rg', 'Cn', 'Nh', 'Fl', 'Mc', 'Lv', 'Ts', 'Og') +SORTED_ELEMENTS = sorted(map(str.lower, ELEMENTS)) +LOWER_PATTERN = re.compile('([a-z]+)') + + +def bsearch(s): + """Return capitalized s if it is a valid symbol of an element, + otherwise return an empty string. + """ + return (lambda i: (i < len(SORTED_ELEMENTS) and SORTED_ELEMENTS[i] == s + and s.capitalize() or ''))(bisect(SORTED_ELEMENTS, s)) + + +def andd(a, b): + """Return sum of a and b if they both evaluate to True, + otherwise return the non-True value. + """ + return a and b and a + b + + +def spongemock(word): + """Return spongemock version of the given lowercase word, + capitalized by symbols of chemical elements. + """ + # For elegance, we don't handle empty strings here. + if len(word) == 1: return bsearch(word) + if len(word) == 2: + return andd(bsearch(word[0]), bsearch(word[1])) or bsearch(word) + return (andd(bsearch(word[:2]), spongemock(word[2:])) + or andd(bsearch(word[0]), spongemock(word[1:]))) + + +def SPoNGeMoCK(string): + """Return spongemock version of the given lowercase string, + capitalized by symbols of chemical elements. + """ + return ''.join(i % 2 and spongemock(s) or s + for i, s in enumerate(LOWER_PATTERN.split(string))) + + +if __name__ == '__main__': stdout.write(SPoNGeMoCK(stdin.read().lower())) -- cgit 1.4.1