aboutsummaryrefslogtreecommitdiff
path: root/challenge-062/sangeet-kar/python/ch-1.py
blob: 3ac4b3768386e8c518785146df1cb6320ab2d9da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys

unique = len(sys.argv) > 1 and sys.argv[1] == '-u'
files = sys.argv[2:] if unique else sys.argv[1:]

mails = []
if files:
    for f in files:
        mails.extend(open(f).read().splitlines())
else:
    mails.extend(sys.stdin.read().splitlines())

mails = [mail.split("@") for mail in mails]

if unique:
    mails = {(name, domain.lower()) for name, domain in mails}

mails = sorted(mails,key=lambda pair: (pair[1].lower(), pair[0]))

print("\n".join(f"{name}@{domain}" for name, domain in mails))