transliterator.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import string

def join_punctuation(seq):
    characters = set(string.punctuation)
    seq = iter(seq)
    current = next(seq)

    for nxt in seq:
        if nxt in characters:
            current += nxt
        else:
            yield current
            current = nxt

    yield current


def main():
    new_line = []

    for word in line.split():
        if word in string.punctuation:
            new_line.append(word)
        elif word in to_translate:
            new_line.append(translate(word))
        else:
            new_line.append(translit(word))

    return ' '.join(join_punctuation(new_line))