misc: small formatting and error checking improvements
This commit is contained in:
parent
369180ff85
commit
03f221a807
@ -1,8 +1,8 @@
|
||||
from textwrap import dedent
|
||||
|
||||
from sqlalchemy import (
|
||||
event,
|
||||
select,
|
||||
event,
|
||||
select,
|
||||
)
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
@ -191,10 +191,10 @@ class WorblehatCli(NumberedCmd):
|
||||
|
||||
def do_exit(self, _: str):
|
||||
if self.sql_session_dirty:
|
||||
if prompt_yes_no('Would you like to save your changes?'):
|
||||
self.sql_session.commit()
|
||||
else:
|
||||
self.sql_session.rollback()
|
||||
if prompt_yes_no('Would you like to save your changes?'):
|
||||
self.sql_session.commit()
|
||||
else:
|
||||
self.sql_session.rollback()
|
||||
exit(0)
|
||||
|
||||
|
||||
|
@ -24,6 +24,9 @@ class Config:
|
||||
]
|
||||
|
||||
def __class_getitem__(cls, name: str) -> Any:
|
||||
if cls._config is None:
|
||||
raise RuntimeError('Configuration not loaded, call Config.load_configuration() first.')
|
||||
|
||||
__config = cls._config
|
||||
for attr in name.split('.'):
|
||||
__config = __config.get(attr)
|
||||
|
Loading…
Reference in New Issue
Block a user