def __getClientOrders(self):
searchMenu = CUI('Client Orders')
self.currentMenu = searchMenu
try:
client_id = int(input('Enter client_id: '))
if not (isinstance(client_id, int) and client_id > 0):
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getAllClientOrders(client_id)
endTime = time.time()
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nRows num: ' + str(len(allRecords)))
searchMenu.addField(
'<Order id> | transaction date | taxes_sum')
for record in allRecords:
searchMenu.addField(
f"<{record[0]}> | {record[1]} | {record[2]}")
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.run('Return to prev menu')
def __getItems(self):
itemsMenu = CUI(self.instance.__name__)
self.itemsCurrentMenu[0] = itemsMenu
try:
count = math.ceil(self.Controller.getCount() / self.per_page)
if self.page < count:
itemsMenu.addField(
'NEXT', lambda: self.__changePageParams(
self.page + 1, self.per_page))
if self.page > 1:
itemsMenu.addField(
'PREV', lambda: self.__changePageParams(
self.page - 1, self.per_page))
entities = self.Controller.getRange(self.page, self.per_page)
itemsMenu.setMsg(f' Page: {self.page}/{count}')
for entity in entities:
if 'name' in self.Controller.getModelKeys():
itemsMenu.addField(f"<{entity.id}> {entity.name}",
lambda id=entity.id: self.__getItem(id))
else:
itemsMenu.addField(f"<{entity.id}>",
lambda id=entity.id: self.__getItem(id))
except Exception as err:
itemsMenu.setMsg(str(err))
itemsMenu.run('Return to main menu')
def __getCategoryProducts(self):
searchMenu = CUI('Product categories')
self.currentMenu = searchMenu
try:
product_id = int(input('Enter product id: '))
if not (isinstance(product_id, int) and product_id > 0):
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getAllProductCategories(
product_id)
endTime = time.time()
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nRows num: ' + str(len(allRecords)))
searchMenu.addField(
f'Product id: {allRecords[0][0]} | Product name: {allRecords[0][1][0]}\nCategories:'
)
for record in allRecords[0][2]:
searchMenu.addField(record)
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.run('Return to prev menu')
def __parsingDataSets(self):
localMenu = CUI('Parsing CSV datasets')
try:
localMenu.addField('Parse set 1', lambda: parse1())
localMenu.addField('Parse set 2', lambda: parse2())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
def __scrapping(self):
localMenu = CUI('Scrapping data')
try:
localMenu.addField('Get data from Amazon',
lambda: getProductsByCategory())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
def __randomizing(self):
localMenu = CUI('Randomizing data')
try:
localMenu.addField('Generate Clients', lambda: generateClients())
localMenu.addField('Generate Orders', lambda: generateOrders())
localMenu.addField('Generate Order-Product relations',
lambda: generateOrderRelation())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
class GraphsView(object):
def __init__(self):
self.CUI = CUI('Graphs menu')
self.CUI.addField('Top 15 categories', lambda: getTop15Categories())
self.CUI.addField('Products per year statistic',
lambda: getManufactureDateStat())
self.CUI.addField('Sale statistics', lambda: getTransactionDateStat())
self.CUI.addField('Get total money per year',
lambda: getTotalMoneyPerYear())
def run(self):
self.CUI.run()
def __getItem(self, id: int):
itemMenu = CUI(f'{self.instance.__name__} menu')
self.itemsCurrentMenu[1] = itemMenu
try:
item = self.Controller.getById(id)
for (key, value
) in self.Controller.getModelEntityMappedKeys(item).items():
itemMenu.addField(str(key) + ' : ' + str(value))
itemMenu.addField('DELETE', lambda: self.__deleteItem(item.id))
itemMenu.addField('UPDATE', lambda: self.__updateItem(item))
itemMenu.addField('Return to prev menu',
lambda: self.__supportCUIFunc())
except Exception as err:
itemMenu.setMsg(str(err))
itemMenu.run(False)
def __getProductsByCostRange(self):
searchMenu = CUI('Products')
self.currentMenu = searchMenu
try:
if self.min is None or self.max is None:
self.min = int(input('Enter min value: '))
self.max = int(input('Enter max value: '))
if not isinstance(self.min, int) and not isinstance(self.max, int) \
and self.max > 0 and 0 < self.min <= self.max:
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getProductsByCostRange(
self.min, self.max)
endTime = time.time()
count = math.ceil(len(allRecords) / self.per_page)
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nPage: ' + str(self.page) + '/' + str(count))
if self.page < math.floor(count):
searchMenu.addField(
'NEXT', lambda: self.__changePageParams(
self.page + 1, self.per_page))
if self.page > 1:
searchMenu.addField(
'PREV', lambda: self.__changePageParams(
self.page - 1, self.per_page))
searchMenu.addField('<ID> | Product name | Product cost')
for record in self.searchController.getProductsByCostRange(
self.min, self.max, self.page, self.per_page):
searchMenu.addField(
f"<{record[0]}> | {record[1]} | {record[2]}")
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.addField('Return to prev menu', lambda: self.__exitMenu())
searchMenu.run(False)
class DataView(object):
def __init__(self):
self.CUI = CUI('New data menu')
self.CUI.addField('Scrapping', lambda: self.__scrapping())
self.CUI.addField('Parsing', lambda: self.__parsingDataSets())
self.CUI.addField('Randomizing', lambda: self.__randomizing())
def run(self):
self.CUI.run()
def __scrapping(self):
localMenu = CUI('Scrapping data')
try:
localMenu.addField('Get data from Amazon',
lambda: getProductsByCategory())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
def __parsingDataSets(self):
localMenu = CUI('Parsing CSV datasets')
try:
localMenu.addField('Parse set 1', lambda: parse1())
localMenu.addField('Parse set 2', lambda: parse2())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
def __randomizing(self):
localMenu = CUI('Randomizing data')
try:
localMenu.addField('Generate Clients', lambda: generateClients())
localMenu.addField('Generate Orders', lambda: generateOrders())
localMenu.addField('Generate Order-Product relations',
lambda: generateOrderRelation())
except Exception as err:
localMenu.setMsg(str(err))
localMenu.run()
class EntityView:
def __init__(self, instance):
self.instance = instance
self.page = 1
self.itemsCurrentMenu = [None, None]
self.per_page = 10
self.CUI = CUI(f"{self.instance.__name__} model menu")
self.Controller = ModelController(instance)
self.CUI.addField(f'Add {self.instance.__name__}',
lambda: self.__addItem())
self.CUI.addField(f'{self.instance.__name__}',
lambda: self.__getItems())
def __supportFillItemFunc(self, key, value, item):
try:
new_value = input(f'Enter new {key} value: ')
if isinstance(new_value, str) and len(new_value) > 0:
if new_value.isdigit():
setattr(item, key, int(new_value))
else:
setattr(item, key, new_value)
else:
raise Exception('Incorrect input')
self.currentMenu.renameField(f'{key}: <{value}>',
f'{key}: <{new_value}>')
except Exception as err:
self.currentMenu.setMsg(str(err))
def __supportFillFunc(self, key, mapped):
try:
value = input(f'Enter new {key} value: ')
old_value = None
if key in mapped and mapped[key] is not None:
old_value = mapped[key]
if isinstance(value, str) and len(value) > 0:
if value.isdigit():
mapped[key] = int(value)
else:
mapped[key] = value
else:
raise Exception('Incorrect input')
if old_value is None:
self.currentMenu.renameField(f'{key}',
f'{key}: <{value}>')
else:
self.currentMenu.renameField(f'{key}: <{old_value}>',
f'{key}: <{value}>')
except Exception as err:
self.currentMenu.setMsg(str(err))
def __fillEntityMenu(self, *args):
self.currentMenu = CUI(f'{self.instance.__name__} fill menu')
try:
if len(args) > 0 and isinstance(args[0], self.instance):
item = args[0]
for (key, value) in self.Controller.getModelEntityMappedKeys(
item).items():
self.currentMenu.addField(
f'{key}: <{value}>',
lambda key=key, value=value: self.
__supportFillItemFunc(key, value, item))
elif len(args) > 0 and isinstance(args[0], dict):
mapped = args[0]
for key in self.Controller.getModelKeys():
mapped[key] = None
self.currentMenu.addField(
f'{key}',
lambda key=key: self.__supportFillFunc(key, mapped))
else:
raise Exception('Invalid arguments')
except Exception as err:
self.currentMenu.setMsg(str(err))
self.currentMenu.run('Save and Return')
def run(self):
self.CUI.run()
def __addItem(self):
try:
mapped = {}
self.__fillEntityMenu(mapped)
exec_str = ''
for value in mapped.values():
if value is None or \
(isinstance(value, str) and
any((char in exec_bad_chars) for char in value)) \
:
raise Exception('Invalid entity fill')
if isinstance(value, str):
exec_str += f"'{value}', "
else:
exec_str += f"{value}, "
exec("self.entity = self.instance(%s)" % exec_str[:-1])
entityId = self.Controller.add(self.entity)
self.CUI.setMsg(f'New Entity created! Id: {entityId}')
except Exception as err:
self.CUI.setMsg(str(err))
def __changePageParams(self, page: int, per_page: int):
self.page = page
self.per_page = per_page
self.itemsCurrentMenu[0].stop()
self.__getItems()
def __getItems(self):
itemsMenu = CUI(self.instance.__name__)
self.itemsCurrentMenu[0] = itemsMenu
try:
count = math.ceil(self.Controller.getCount() / self.per_page)
if self.page < count:
itemsMenu.addField(
'NEXT', lambda: self.__changePageParams(
self.page + 1, self.per_page))
if self.page > 1:
itemsMenu.addField(
'PREV', lambda: self.__changePageParams(
self.page - 1, self.per_page))
entities = self.Controller.getRange(self.page, self.per_page)
itemsMenu.setMsg(f' Page: {self.page}/{count}')
for entity in entities:
if 'name' in self.Controller.getModelKeys():
itemsMenu.addField(f"<{entity.id}> {entity.name}",
lambda id=entity.id: self.__getItem(id))
else:
itemsMenu.addField(f"<{entity.id}>",
lambda id=entity.id: self.__getItem(id))
except Exception as err:
itemsMenu.setMsg(str(err))
itemsMenu.run('Return to main menu')
def __updateItem(self, item):
self.__fillEntityMenu(item)
self.Controller.update(item)
self.itemsCurrentMenu[1].stop()
self.__getItem(item.id)
def __deleteItem(self, id: int):
self.Controller.delete(id)
self.itemsCurrentMenu[1].stop()
self.__supportCUIFunc()
def __supportCUIFunc(self):
self.itemsCurrentMenu[1].stop()
self.__changePageParams(self.page, self.per_page)
def __getItem(self, id: int):
itemMenu = CUI(f'{self.instance.__name__} menu')
self.itemsCurrentMenu[1] = itemMenu
try:
item = self.Controller.getById(id)
for (key, value
) in self.Controller.getModelEntityMappedKeys(item).items():
itemMenu.addField(str(key) + ' : ' + str(value))
itemMenu.addField('DELETE', lambda: self.__deleteItem(item.id))
itemMenu.addField('UPDATE', lambda: self.__updateItem(item))
itemMenu.addField('Return to prev menu',
lambda: self.__supportCUIFunc())
except Exception as err:
itemMenu.setMsg(str(err))
itemMenu.run(False)
class SearchView:
def __init__(self):
self.currentMenu = None
self.page = 1
self.per_page = 10
self.min = None
self.max = None
self.CUI = CUI("Search menu")
self.searchController = SearchController()
self.CUI.addField('Search products cost by cost range',
lambda: self.__getProductsByCostRange())
self.CUI.addField('Search all client orders',
lambda: self.__getClientOrders())
self.CUI.addField('Search all product categories',
lambda: self.__getCategoryProducts())
def run(self):
self.CUI.run()
def __changePageParams(self, page: int, per_page: int):
self.page = page
self.per_page = per_page
self.currentMenu.stop()
self.__getProductsByCostRange()
def __exitMenu(self):
self.min = None
self.max = None
self.page = 1
self.per_page = 10
self.currentMenu.stop()
def __getProductsByCostRange(self):
searchMenu = CUI('Products')
self.currentMenu = searchMenu
try:
if self.min is None or self.max is None:
self.min = int(input('Enter min value: '))
self.max = int(input('Enter max value: '))
if not isinstance(self.min, int) and not isinstance(self.max, int) \
and self.max > 0 and 0 < self.min <= self.max:
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getProductsByCostRange(
self.min, self.max)
endTime = time.time()
count = math.ceil(len(allRecords) / self.per_page)
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nPage: ' + str(self.page) + '/' + str(count))
if self.page < math.floor(count):
searchMenu.addField(
'NEXT', lambda: self.__changePageParams(
self.page + 1, self.per_page))
if self.page > 1:
searchMenu.addField(
'PREV', lambda: self.__changePageParams(
self.page - 1, self.per_page))
searchMenu.addField('<ID> | Product name | Product cost')
for record in self.searchController.getProductsByCostRange(
self.min, self.max, self.page, self.per_page):
searchMenu.addField(
f"<{record[0]}> | {record[1]} | {record[2]}")
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.addField('Return to prev menu', lambda: self.__exitMenu())
searchMenu.run(False)
def __getClientOrders(self):
searchMenu = CUI('Client Orders')
self.currentMenu = searchMenu
try:
client_id = int(input('Enter client_id: '))
if not (isinstance(client_id, int) and client_id > 0):
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getAllClientOrders(client_id)
endTime = time.time()
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nRows num: ' + str(len(allRecords)))
searchMenu.addField(
'<Order id> | transaction date | taxes_sum')
for record in allRecords:
searchMenu.addField(
f"<{record[0]}> | {record[1]} | {record[2]}")
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.run('Return to prev menu')
def __getCategoryProducts(self):
searchMenu = CUI('Product categories')
self.currentMenu = searchMenu
try:
product_id = int(input('Enter product id: '))
if not (isinstance(product_id, int) and product_id > 0):
raise Exception('Invalid input')
startTime = time.time()
allRecords = self.searchController.getAllProductCategories(
product_id)
endTime = time.time()
searchMenu.setMsg('\nElapsed time: ' +
str(endTime - startTime)[:9] + 's'
'\nRows num: ' + str(len(allRecords)))
searchMenu.addField(
f'Product id: {allRecords[0][0]} | Product name: {allRecords[0][1][0]}\nCategories:'
)
for record in allRecords[0][2]:
searchMenu.addField(record)
except Exception as err:
searchMenu.setMsg(str(err))
searchMenu.run('Return to prev menu')
from db import session, recreate_database
from views.modelView import EntityView
from CUI.constructor import CUI
from models.client import Client
from models.order import Order
from models.category import Category
from models.product import Product
from views.graphsView import GraphsView
from views.dataView import DataView
from views.searchView import SearchView
if __name__ == '__main__':
#recreate_database()
cui = CUI('CUI')
cui.addMenu('Models')
cui.addField('Clients', lambda: EntityView(Client).run())
cui.addField('Orders', lambda: EntityView(Order).run())
cui.addField('Categories', lambda: EntityView(Category).run())
cui.addField('Products', lambda: EntityView(Product).run())
cui.finishMenu()
cui.addField('Graphs', lambda: GraphsView().run())
cui.addField('Data', lambda: DataView().run())
cui.addField('Search', lambda: SearchView().run())
cui.run()
session.close()