Initial version

menus/main.py

@@ -0,0 +1,76 @@
+from ursina import *
+import pypresence, asyncio, json
+from utils.utils import FakePyPresence
+from utils.constants import discord_presence_id
+
+class MenuButton(Button):
+    def __init__(self, text='', **kwargs):
+        super().__init__(text, scale=(.25, .075), highlight_color=color.azure, **kwargs)
+
+        for key, value in kwargs.items():
+            setattr(self, key ,value)
+
+class Main():
+    def __init__(self, pypresence_client=None) -> None:
+        self.pypresence_client = pypresence_client
+
+        with open("settings.json", "r") as file:
+            self.settings_dict = json.load(file)
+
+        if self.settings_dict.get('discord_rpc', True):
+            if self.pypresence_client == None: # Game has started
+                try:
+                    asyncio.get_event_loop()
+                except:
+                    asyncio.set_event_loop(asyncio.new_event_loop())
+                try:
+                    self.pypresence_client = pypresence.Presence(discord_presence_id)
+                    self.pypresence_client.connect()
+                    self.pypresence_client.start_time = time.time()
+                except:
+                    self.pypresence_client = FakePyPresence()
+                    self.pypresence_client.start_time = time.time()
+
+            elif isinstance(self.pypresence_client, FakePyPresence): # the user has enabled RPC in the settings in this session.
+                # get start time from old object
+                start_time = copy.deepcopy(self.pypresence_client.start_time)
+                try:
+                    self.pypresence_client = pypresence.Presence(discord_presence_id)
+                    self.pypresence_client.connect()
+                    self.pypresence_client.start_time = start_time
+                except:
+                    self.pypresence_client = FakePyPresence()
+                    self.pypresence_client.start_time = start_time
+
+        else: # game has started, but the user has disabled RPC in the settings.
+            self.pypresence_client = FakePyPresence()
+            self.pypresence_client.start_time = time.time()
+
+        self.pypresence_client.update(state='In Main Menu', details='In Main Menu')
+
+        button_spacing = .075 * 1.25
+        base_y = -2
+        self.menu_parent = Entity(parent=camera.ui, y=.15)
+        self.main_menu = Entity(parent=self.menu_parent)
+
+        self.title_label = Text("Aim Trainer", parent=self.main_menu, y=-0.25 * button_spacing, scale=3, x=-.2)
+        self.play_button = MenuButton('Play', on_click=Func(self.play), parent=self.main_menu, y=-2 * button_spacing)
+        self.settings_button = MenuButton('Settings', on_click=Func(self.settings), parent=self.main_menu, y=-3 * button_spacing)
+        self.quit_button = MenuButton('Quit', on_click=Sequence(Wait(.01), Func(application.quit)), parent=self.main_menu, y=-4 * button_spacing)
+
+    def play(self):
+        self.hide()
+        from game.game import Game
+        Game(self.pypresence_client)
+
+    def settings(self):
+        self.hide()
+        from menus.settings import Settings
+        Settings(self.pypresence_client)
+
+    def hide(self):
+        destroy(self.play_button)
+        destroy(self.settings_button)
+        destroy(self.quit_button)
+        destroy(self.main_menu)
+        destroy(self.menu_parent)