From 1b09a904cb2a968369a71d40e6fb74408a2fedfc Mon Sep 17 00:00:00 2001
From: h7x4 <h7x4@nani.wtf>
Date: Tue, 27 Jan 2026 19:40:17 +0900
Subject: [PATCH] menus/mainmenu: register sql session in menu

---
 dibbler/menus/mainmenu.py   | 4 ++++
 dibbler/subcommands/loop.py | 1 -
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/dibbler/menus/mainmenu.py b/dibbler/menus/mainmenu.py
index 8ed9925..d822869 100644
--- a/dibbler/menus/mainmenu.py
+++ b/dibbler/menus/mainmenu.py
@@ -3,6 +3,7 @@ import os
 import random
 import sys
 
+from sqlalchemy.orm import Session
 
 from .buymenu import BuyMenu
 from .faq import FAQMenu
@@ -19,6 +20,9 @@ def restart():
 
 
 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:
         mv = in_str.split()
         if len(mv) == 2 and mv[0].isdigit():
diff --git a/dibbler/subcommands/loop.py b/dibbler/subcommands/loop.py
index 10f0d9f..452320d 100755
--- a/dibbler/subcommands/loop.py
+++ b/dibbler/subcommands/loop.py
@@ -52,7 +52,6 @@ def main(sql_session: Session):
         set_signal_handler(SIGTSTP, SIG_IGN)
 
     main = MainMenu(
-        "Dibbler main menu",
         sql_session,
         items=[
             BuyMenu(sql_session),