Sklad.py

import re

#Вывод меню программы на экран
def main():
    print('''[1] - Добавить запись в базу
[2] - Редактирование записи
[3] - Удалить запись из базы
[4] - Вывести все записи базы на экран
[5] - Очистка всей бызы
[6] - Запись базы в файл
[7] - Загрузка базы из файла
[m] - Меню программы
[q] - Выход из программы''')

#Вывод записи на экран
def base_list(key):
    print(key, sklad[key][0].ljust(22, ' '),
          sklad[key][1],
          str(sklad[key][2]).rjust(5, ' '),
          (str(sklad[key][3])+' грн').rjust(9, ' '),
          str(sklad[key][2]*sklad[key][3])+' грн',
          sep=' | ')

sklad = {}

ext = r"[qQйЙ]"
menu = r"[mMьЬ]"

while True:
    select = input('--> ')

    #[mMьЬ] - Меню программы
    if re.search(menu, select):
        main()

    #[qQйЙ] - Выход из программы
    elif re.search(ext, select):
        break

    #[1] - Добавление товара в базу
    elif select == '1':
        key = input('Номер: ')
        if key in sklad.keys():
            print('Запись с таким номером существует!')
        else:
            name = input('Название товара: ')
            edinica = input('Единица измерения: ')
            kol_vo = float(input('Количество: '))
            cena = float(input('Цена за единицу: '))
            sklad.setdefault(key, [name, edinica, kol_vo, cena])

    #[2] - Редактирование записи
    elif select == '2':
        print('\nРедактирование записи в базе\n')
        key = input('Введите номер записи: ')
        if key in sklad.keys():
            base_list(key)
        else:
            print('Записи с таким номером нет в базе!')

    #[3] - Удаление товара из базы
    elif select == '3':
        key = input('Введите номер записи: ')
        if key != '':
            if key in sklad.keys():
                sklad.pop(key)
                print('Запись удалена!')
            else:
                print('Записи с таким номером нет в базе!')
        print()

    #[4] - Вывод всех записей базы на экран
    elif select == '4':
        if len(sklad) != 0:
            for key in sklad:
                base_list(key)
        else:
            print('База данных пуста!')

    #[5] - Очистка всей базы
    elif select == '5':
        sklad.clear()
        print('Все записи базы удалены!')

    #[6] - Запись базы в файл
    elif select == '6':
        base = open('base.txt', 'w')
        for key in sklad:
            sklad_base = (key+';'+sklad[key][0]+';'+sklad[key][1]+';'+
                          str(sklad[key][2])+';'+str(sklad[key][3])+
                          '|')
            base.write(sklad_base)
        base.close()
        n = len(sklad)
        print('В файл записано записей: '+str(n))

    #[7] - Загрузка базы из файла
    elif select == '7':
        sklad.clear()
        base = open('base.txt', 'r')

        for n in base:
            lst = n.split('|')
            for i in lst:
                if i != '':
                    x = i.split(';')
                    sklad.setdefault(x[0], [x[1],
                                            x[2],
                                            float(x[3]),
                                            float(x[4])])
        base.close()
        n = len(sklad)
        print('\nЗагружено записей: '+str(n)+'\n')
        for key in sklad:
            base_list(key)