From 1b9381ac8f9b9ff7bc035b89e399ea5b12ea8c11 Mon Sep 17 00:00:00 2001
From: hyperbel <hyperbel@users.noreply.github.com>
Date: Mon, 30 Jan 2023 14:59:55 +0100
Subject: [PATCH] use args on run.sh

---
 main.py   | 4 ++--
 run.sh    | 2 +-
 shell.nix | 6 ++++++
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index c0205d2..fcde783 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 import argparse
-import ArgumentParser
+from argparse import ArgumentParser
 from server import server
 
 def main():
@@ -27,4 +27,4 @@ def main():
         parser.print_help()
 
 if __name__ == "__main__":
-    exit(main())
\ No newline at end of file
+    exit(main())
diff --git a/run.sh b/run.sh
index d487422..0487431 100755
--- a/run.sh
+++ b/run.sh
@@ -7,4 +7,4 @@ cat ./Config/party.sql | sqlite3 party.db
 
 export FLASK_APP=main.py
 
-sudo python3 main.py
+sudo python3 main.py $@
diff --git a/shell.nix b/shell.nix
index c9f813e..08eeedb 100644
--- a/shell.nix
+++ b/shell.nix
@@ -14,4 +14,10 @@ pkgs.mkShell {
     pkgs.tmux
     pkgs.tree
   ];
+
+  shellHook = ''
+    export FLASK_APP=main.py
+    export FLASK_ENV=development
+    export FLASK_RUN_PORT=5000
+    '';
 }