def test_step09_work_address_not_eq(self):
from work_address import WorkAddress
first = WorkAddress("Adam Adamski", "Warszawa", "ul. Domaniewska",
"6/66", "Mordor sp. z o.o.")
second = WorkAddress("Adam Adamski", "Warszawa", "ul. Domaniewska",
"6/66", "Hobbit sp. z o.o.")
self.assertNotEqual(first, second)
def create_from_csv(list_name, csv_path):
address_book = AddressBook(list_name)
list_of_addresses = []
with open(csv_path) as file:
for line in file:
list_of_addresses.append(line[:-1].split(","))
list_of_addresses.pop(0)
for address in list_of_addresses:
if address[-1] == "":
address_book.addresses.append(Address(*address[:-1]))
else:
address_book.addresses.append(WorkAddress(*address))
return address_book
def create_from_csv(cls, list_name, csv_path):
address_book = cls(list_name)
with open(csv_path) as csvfile:
address_reader = csv.reader(csvfile)
next(address_reader) # skip heades
for data in address_reader:
company = data[4]
if company:
address_book.add_address(
WorkAddress(data[0], data[1], data[2], data[3],
data[4], data[5]))
else:
address_book.add_address(
Address(data[0], data[1], data[2], data[3], data[4]))
return address_book
def create_from_csv(list_name, csv_path):
database_list = AddressBook.load_items_from_file(csv_path)
database_list.remove(database_list[0])
book = AddressBook(list_name)
for element in database_list:
if element[4] == '':
address = Address(element[0], element[1], element[2], element[3])
book.add_address(address)
else:
address = WorkAddress(element[0], element[1], element[2], element[3], element[4])
book.add_address(address)
return book
def create_from_csv(last_name, csv_path):
new_address_book = AddressBook(last_name)
with open(csv_path) as source:
addresses_list = csv.DictReader(source)
for address in addresses_list:
if address['company'] == '':
new_address = Address(address["person"], address["city"], address["street"], address["house_no"])
new_address_book.add_address(new_address)
else:
new_address = WorkAddress(address["person"], address["city"], address["street"],
address["house_no"], address["company"])
new_address_book.add_address(new_address)
return new_address_book
def create_address_book(self):
from address_book import AddressBook
from address import Address
from work_address import WorkAddress
expected = AddressBook("expected")
expected.add_address(
Address("Jane Weaver", "Tugusirna", "Rowland", "1877/2"))
expected.add_address(
Address("Rebecca Cunningham", "Murzuq", "Heffernan", "2/3"))
expected.add_address(
Address("Steven Pierce", "Matsena", "Mariners Cove", "153/4"))
expected.add_address(
WorkAddress("Jerzy Mardaus", "Kraków", "Ślusarska", "9/1",
"Codecool Poland Sp. z o.o."))
expected.add_address(
Address("Betty Jenkins", "Tirmiz", "Hollow Ridge", "011/5"))
return expected
def create_from_csv(list_name, csv_path):
new_book = AddressBook(list_name)
with open(csv_path, 'r') as f:
next(f)
reader = csv.reader(f, delimiter=',')
for line in reader:
person = line[0]
city = line[1]
street = line[2]
house_no = line[3]
company = line[4]
if company == '':
address = Address(person, city, street, house_no)
else:
address = WorkAddress(person, city, street, house_no,
company)
new_book.add_address(address)
return new_book
def get_work_address(self):
from work_address import WorkAddress
return WorkAddress("Adam Adamski", "Warszawa", "ul. Domaniewska",
"6/66", "Mordor sp. z o.o.")
from work import Work
from work_address import WorkAddress
from workcontact import WorkContact
if __name__ == "__main__":
while True:
''' START WITH INPUT '''
COMMAND = input(
'What would you like to do (add, list, exit)? ').lower()
if COMMAND == 'exit':
print('Deuce deuce!')
break
elif COMMAND == 'add':
ADDRESS_BOOK = AddressBook()
WORK_ADDRESS = WorkAddress()
PER_NAME = input('What is your name?').upper()
PER_PHONE = input('Phone: ')
ADDRESS_BOOK.add_contact(
Contact(name=PER_NAME,
phone=PER_PHONE,
home_address=Address(
street=input('Street number and street: '),
city=input('City: ').upper(),
state=input('State: ').upper(),
zipcode=input('Zipcode: '))))
while True:
SUB_COMND = input(
'would you like to add your work information yes or no? '
).upper()
def add_address(address_book):
address = View.add_address()
if len(address) == 4:
address_book.add_address(Address(*address))
else:
address_book.add_address(WorkAddress(*address))