back buttons for assistant

UltroidVenom · Feb 26, 2021 · bfe741a · bfe741a
1 parent 34db188
commit bfe741a
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 12 deletions.
diff --git a/assistant/api_setter.py b/assistant/api_setter.py
@@ -15,8 +15,10 @@
 async def apiset(event):
     await event.edit(
         "Choose which API you want to set.",
-        buttons=[[Button.inline("Remove.bg", data="rmbg")]],
-    )
+        buttons=[
+            [Button.inline("Remove.bg", data="rmbg")],
+            [custom.Button.inline("« Back", data="setter")]
+            ])
 
 
 # remove.bg api

diff --git a/assistant/customvars.py b/assistant/customvars.py
@@ -27,6 +27,7 @@ async def alvcs(event):
             [Button.inline("Alive Text", data="alvtx")],
             [Button.inline("Alive Media", data="alvmed")],
             [Button.inline("Delete Alive Media", data="delmed")],
+            [Button.inline("« Back", data="allcstms")]
         ],
     )
 
@@ -103,6 +104,7 @@ async def alvcs(event):
             [Button.inline("PM Text", data="pmtxt")],
             [Button.inline("Pm Media", data="pmmed")],
             [Button.inline("Delete PM Media", data="delpmmed")],
+            [Button.inline("« Back", data="allcstms")]
         ],
     )
 

diff --git a/assistant/othervars.py b/assistant/othervars.py
@@ -19,8 +19,8 @@ async def otvaar(event):
             [Button.inline("Tag Logger", data="taglog")],
             [Button.inline("PM Permit", data="pmset")],
             [Button.inline("SuperFban", data="sfban")],
-        ],
-    )
+            [Button.inline("« Back", data="setter")]
+        ])
 
 
 @callback("taglog")
@@ -52,6 +52,7 @@ async def pmset(event):
         buttons=[
             [Button.inline("Turn PMPermit On", data="pmon")],
             [Button.inline("Turn PMPermit Off", data="pmoff")],
+            [Button.inline("« Back", data="otvars")]
         ],
     )
 
@@ -80,6 +81,7 @@ async def sfban(event):
         buttons=[
             [Button.inline("FBan Group", data="sfgrp")],
             [Button.inline("Exclude Feds", data="sfexf")],
+            [Button.inline("« Back", data="otvars")]
         ],
     )
 

diff --git a/assistant/start.py b/assistant/start.py
@@ -46,12 +46,26 @@ async def ultroid(event):
         event.chat_id,
         f"Hi {OWNER_NAME}. Please browse through the options",
         buttons=[
-            [custom.Button.inline("Settings ⚙️", data="setter")],
-            [custom.Button.inline("Stats", data="stat")],
-            [custom.Button.inline("Broadcast", data="bcast")],
+            [Button.inline("Settings ⚙️", data="setter")],
+            [Button.inline("Stats", data="stat")],
+            [Button.inline("Broadcast", data="bcast")],
         ],
     )
 
+# aah, repeat the codes..
+@callback("mainmenu")
+@owner
+async def ultroid(event):
+    if event.is_group:
+        return
+    await event.edit(
+        f"Hi {OWNER_NAME}. Please browse through the options",
+        buttons=[
+            [Button.inline("Settings ⚙️", data="setter")],
+            [Button.inline("Stats", data="stat")],
+            [Button.inline("Broadcast", data="bcast")],
+        ],
+    )
 
 @callback("stat")
 @owner
@@ -106,9 +120,10 @@ async def setting(event):
     await event.edit(
         "Choose from the below options -",
         buttons=[
-            [custom.Button.inline("Customisations", data="allcstms")],
-            [custom.Button.inline("API Keys", data="apiset")],
-            [custom.Button.inline("Other Vars.", data="otvars")],
+            [Button.inline("Customisations", data="allcstms")],
+            [Button.inline("API Keys", data="apiset")],
+            [Button.inline("Other Vars.", data="otvars")],
+            [Button.inline("« Back", data="mainmenu")]
         ],
     )
 
@@ -118,6 +133,7 @@ async def all(event):
     await event.edit(
         "All Customisable Stuff.",
         buttons=[
-            [custom.Button.inline("Alive", data="alvcstm")],
-            [custom.Button.inline("PM Permit", data="pmcstm")]
+            [Button.inline("Alive", data="alvcstm")],
+            [Button.inline("PM Permit", data="pmcstm")],
+            [Button.inline("« Back", data="setter")]
             ])