class FileMailListAdapterTest(unittest.TestCase):
"""docstring for FileMailListAdapterTest"""
def setUp(self):
file = open("test.txt", "w")
contents = "Pepa - pepa@pepa"
file.write(contents)
file.close()
self.obj = FileMailListAdapter("test.txt")
def test_make_subscribers(self):
expected = {}
expected["Pepa"] = "pepa@pepa"
self.assertEqual(expected, self.obj.make_subscribers())
def tearDown(self):
call("rm " + "test.txt", shell=True)
def main():
print("Hello Stranger! This is a cutting-edge, console-based mail-list! Type help, to see a list of commands.")
while True:
command = input(">")
command = command.split()
if command[0] == "help":
call("python help.py", shell=True)
if command[0] == "show_lists":
if len(command) == 1:
lists = get_lists()
if not len(lists) == 0:
for i in range(len(lists)):
lists[i] = lists[i][0:len(lists[i]) - 4]
print("[" + str(i + 1) + "]" + " - " + lists[i])
else:
print ("There are no email lists!")
if command[0] == "show_list":
lists = get_lists()
i = int(command[1])
try:
l = open("%s" % (lists[i - 1]), "r")
except IndexError:
print("List with unique identifier %s was not found!" %(i))
else:
list = l.read()
list = list.split('\n')
if not list == ['']:
for i in range(len(list)):
print("[" + str(i + 1) + "]" + " " + list[i])
else:
print ("The mail list is empty!")
if command[0] == "add":
lists = get_lists()
i = int(command[1]) - 1
name = input("Name: ")
email = input("Email ")
f = FileMailListAdapter(lists[i])
mail = f.getMail()
emails = mail.get_emails()
if emails.count(email) == 0:
mail.add_subscriber(name, email)
adapter = MailListFileAdapter(mail)
adapter.save()
else:
print("A person with the given email <%s> is already in the list!" %(email))
if command[0] == "update_subscriber":
lists = get_lists()
i = int(command[1])
j = int(command[2])
if i <= len(lists):
f = FileMailListAdapter(lists[i - 1])
mail = f.getMail()
subscribers = mail.get_subscribers()
if j <= len(subscribers):
print("Updating: %s - %s \n Pres enter if you want the field to remain the same" %(subscribers[j - 1][0],subscribers[j - 1][1]))
old_name = subscribers[j - 1][0]
old_email = subscribers[j - 1][1]
name = input("enter new name> ")
email = input("enter new email> ")
if name != "" and email != "":
mail.update_subscriber(old_name, name, email)
elif name == "" and email != "":
mail.update_subscriber(old_name, old_name, email)
elif name != "" and email == "":
mail.update_subscriber(old_name, name, old_email)
else:
mail.update_subscriber(old_name,old_name,old_email)
adapter = MailListFileAdapter(mail)
adapter.save()
else:
print ("Subscriber with identifider <" + str(j) + "> not found in the list." )
else:
print("List with unique identifier <" + str(i) + "> was not found.")
if command[0] == "remove_subscriber":
lists = get_lists()
i = int(command[1])
j = int(command[2])
if i <= len(lists):
f = FileMailListAdapter(lists[i - 1])
mail = f.getMail()
subscribers = mail.get_subscribers()
if j <= len(subscribers):
mail.remove_subscriber(subscribers[j - 1][0])
adapter = MailListFileAdapter(mail)
adapter.save()
else:
print ("Subscriber with identifider <" + str(j) + "> not found in the list.")
else:
print("List with unique identifier <"+ str(i) + "> was not found")
if command[0] == "exit":
break
if command[0] == "create":
lists = get_lists()
if command[1] + ".txt" in lists:
print ("A list with name <" + command[1] + "> already exists!")
else:
mail = MailList(command[1] + ".txt")
f = MailListFileAdapter(mail)
f.save()
print ("New list <" + command[1] + "> was created")
if command[0] == "update":
lists = get_lists()
i = int(command[1])
if i > len(lists):
print("List with unique identifier <" + str(i) + "> \
was not found")
else:
f = FileMailListAdapter(lists[i - 1])
mail = f.getMail()
mail.set_name(command[2])
adapter = MailListFileAdapter(mail)
adapter.save()
print("Updated " + lists[i - 1] + " to " + command[2])
if command[0] == "delete":
lists = get_lists()
i = int(command[1]) - 1
if i <= len(lists):
print ("Are you sure you want to delete " + lists[i])
choice = input("(Y/N)>")
if choice == 'Y':
if i <= len(lists):
call("rm " + lists[i], shell=True)
else:
print("You crazy bastard. Stop playing with fire!")
else:
print("List with unique identifier <" + str(i) + "> \
was not found.")
if command[0] == "search_email":
lists = get_lists()
result = []
for i in range(len(lists)):
f = FileMailListAdapter(lists[i])
mail = f.getMail()
subscribers = mail.get_subscribers()
for j in range(len(subscribers)):
if command[1] == subscribers[j][1]:
result.append(lists[i][:len(lists[i]) - 4:])
if len(result) == 0:
print("<" + command[1] + "> was not found in the \
current mailing lists.")
else:
for k in range(len(result)):
print("[" + str(k + 1) + "] " + result[k])
if command[0] == "merge_lists":
lists = get_lists()
i = int(command[1]) - 1
j = int(command[2]) - 1
new_list = command[3]
try:
list1 = lists[i]
list2 = lists[j]
except IndexError:
print("Can`t find lists with the given number")
else:
print("Merged lists <%s> and <%s> into <%s>" %(lists[i][:len(lists[i]) - 4:],lists[j][:len(lists[j]) - 4:],new_list))
new_mail = MailList(new_list + ".txt")
f = MailListFileAdapter(new_mail)
f.save()
m1 = FileMailListAdapter(lists[i])
mail1 = m1.getMail()
subs1 = mail1.get_subscribers()
emails1 = mail1.get_emails()
for name in subs1:
new_mail.add_subscriber(name[0],name[1])
m2 = FileMailListAdapter(lists[j])
mail2 = m2.getMail()
subs2 = mail2.get_subscribers()
for name in subs2:
if emails1.count(name[1]) == 0:
new_mail.add_subscriber(name[0],name[1])
adapter = MailListFileAdapter(new_mail)
adapter.save()
if command[0] == "export":
lists = get_lists()
i = int(command[1]) - 1
try:
f = FileMailListAdapter(lists[i])
except IndexError:
print("List with unique identifier <" + str(i) + "> was not found.")
else:
print("Exported <%s> to <%s.json>" %(lists[i][:len(lists[i]) - 4:],lists[i][:len(lists[i]) - 4:]))
mail = f.getMail()
subscribers = mail.return_subscribers()
json_file = json.dumps(subscribers,sort_keys=True,indent=4,ensure_ascii=False)
print(json_file)
json_list = open("%s.json" %(lists[i][:len(lists[i]) - 4:]),"w")
json_list.write("%s" %(json.loads(json_file)))
def setUp(self):
file = open("test.txt", "w")
contents = "Pepa - pepa@pepa"
file.write(contents)
file.close()
self.obj = FileMailListAdapter("test.txt")
