menus/mainmenu: register sql session in menu
This commit is contained in:
@@ -3,6 +3,7 @@ import os
|
|||||||
import random
|
import random
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
|
||||||
from .buymenu import BuyMenu
|
from .buymenu import BuyMenu
|
||||||
from .faq import FAQMenu
|
from .faq import FAQMenu
|
||||||
@@ -19,6 +20,9 @@ def restart():
|
|||||||
|
|
||||||
|
|
||||||
class MainMenu(Menu):
|
class MainMenu(Menu):
|
||||||
|
def __init__(self, sql_session: Session, **kwargs):
|
||||||
|
super().__init__("Dibbler main menu", sql_session, **kwargs)
|
||||||
|
|
||||||
def special_input_choice(self, in_str: str) -> bool:
|
def special_input_choice(self, in_str: str) -> bool:
|
||||||
mv = in_str.split()
|
mv = in_str.split()
|
||||||
if len(mv) == 2 and mv[0].isdigit():
|
if len(mv) == 2 and mv[0].isdigit():
|
||||||
|
|||||||
@@ -52,7 +52,6 @@ def main(sql_session: Session):
|
|||||||
set_signal_handler(SIGTSTP, SIG_IGN)
|
set_signal_handler(SIGTSTP, SIG_IGN)
|
||||||
|
|
||||||
main = MainMenu(
|
main = MainMenu(
|
||||||
"Dibbler main menu",
|
|
||||||
sql_session,
|
sql_session,
|
||||||
items=[
|
items=[
|
||||||
BuyMenu(sql_session),
|
BuyMenu(sql_session),
|
||||||
|
|||||||
Reference in New Issue
Block a user