h7x4
18a1667b7b
- Add missing flushes to cli - Update README TODOs - Add some documentation comments - Rename AdvancedOptions -> AdvancedOptionsCli
111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from worblehat.cli.prompt_utils import (
|
|
InteractiveItemSelector,
|
|
NumberedCmd,
|
|
format_date,
|
|
prompt_yes_no,
|
|
)
|
|
from worblehat.models import Bookcase, BookcaseShelf
|
|
|
|
class AdvancedOptionsCli(NumberedCmd):
|
|
def __init__(self, sql_session: Session):
|
|
super().__init__()
|
|
self.sql_session = sql_session
|
|
|
|
|
|
def do_add_bookcase(self, _: str):
|
|
while True:
|
|
name = input('Name of bookcase> ')
|
|
if name == '':
|
|
print('Error: name cannot be empty')
|
|
continue
|
|
|
|
if self.sql_session.scalars(
|
|
select(Bookcase)
|
|
.where(Bookcase.name == name)
|
|
).one_or_none() is not None:
|
|
print(f'Error: a bookcase with name {name} already exists')
|
|
continue
|
|
|
|
break
|
|
|
|
description = input('Description of bookcase> ')
|
|
if description == '':
|
|
description = None
|
|
|
|
bookcase = Bookcase(name, description)
|
|
self.sql_session.add(bookcase)
|
|
self.sql_session.flush()
|
|
|
|
|
|
def do_add_bookcase_shelf(self, arg: str):
|
|
bookcase_selector = InteractiveItemSelector(
|
|
cls = Bookcase,
|
|
sql_session = self.sql_session,
|
|
)
|
|
bookcase_selector.cmdloop()
|
|
bookcase = bookcase_selector.result
|
|
|
|
while True:
|
|
column = input('Column> ')
|
|
try:
|
|
column = int(column)
|
|
except ValueError:
|
|
print('Error: column must be a number')
|
|
continue
|
|
break
|
|
|
|
while True:
|
|
row = input('Row> ')
|
|
try:
|
|
row = int(row)
|
|
except ValueError:
|
|
print('Error: row must be a number')
|
|
continue
|
|
break
|
|
|
|
if self.sql_session.scalars(
|
|
select(BookcaseShelf)
|
|
.where(
|
|
BookcaseShelf.bookcase == bookcase,
|
|
BookcaseShelf.column == column,
|
|
BookcaseShelf.row == row,
|
|
)
|
|
).one_or_none() is not None:
|
|
print(f'Error: a bookshelf in bookcase {bookcase.name} with position c{column}-r{row} already exists')
|
|
return
|
|
|
|
description = input('Description> ')
|
|
if description == '':
|
|
description = None
|
|
|
|
shelf = BookcaseShelf(
|
|
row,
|
|
column,
|
|
bookcase,
|
|
description,
|
|
)
|
|
self.sql_session.add(shelf)
|
|
self.sql_session.flush()
|
|
|
|
|
|
def do_done(self, _: str):
|
|
return True
|
|
|
|
|
|
funcs = {
|
|
1: {
|
|
'f': do_add_bookcase,
|
|
'doc': 'Add bookcase',
|
|
},
|
|
2: {
|
|
'f': do_add_bookcase_shelf,
|
|
'doc': 'Add bookcase shelf',
|
|
},
|
|
9: {
|
|
'f': do_done,
|
|
'doc': 'Done',
|
|
},
|
|
} |