def move_document(doc_number, new_shelf):
for shelf in directories.keys():
if doc_number in directories[shelf] and new_shelf in directories.keys():
directories[shelf].remove(doc_number)
directories[new_shelf].append(doc_number)
return f'Документ номер {doc_number} перемещен на полку {new_shelf}'
else:
return 'Документа или полки с такими номерами не существует'
def test_absence_in_directories(self):
del_a_doc(documents, directories, old_number)
values_list = [directories.get(key) for key in directories.keys()]
new_values_list = []
for values in values_list:
new_values_list.extend(values)
self.assertNotIn(old_number, new_values_list)
def test_existence_in_directories(self):
add_a_doc(documents, directories, new_type, new_number, new_owner,
new_shelf_number)
values_list = [directories.get(key) for key in directories.keys()]
new_values_list = []
for values in values_list:
new_values_list.extend(values)
self.assertIn(new_number, new_values_list)
def add_doc(self, type_doc, number, name, shelf):
self.type_doc = type_doc
self.number = number
self.name = name
self.shelf = shelf
if self.shelf not in directories.keys(): # проверяем на наличие полки
print('Полки с таким номером не существует')
return None
new_human = {
} # создаем словарь с новыми данными и добавляем в общий каталог
new_human["type"] = self.type_doc
new_human["number"] = self.number
new_human["name"] = self.name
documents.append(new_human)
directories[self.shelf] += [
self.number
] # добавляем документ на необходимую полку
print(f'В каталог добавлена новая запись {new_human}')
print(f'Документ №{self.number} добавлен на полку {self.shelf}')
return new_human
def add_new_shelf(new_shelf):
if new_shelf in directories.keys():
return f'Полка {new_shelf} уже есть'
else:
directories[new_shelf] = []
return f'Полка {new_shelf} создана'