Try environment variables

TJCSec · Dec 1, 2024 · e8276c1 · e8276c1
1 parent 83ee8d8
commit e8276c1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/update_website.yaml b/.github/workflows/update_website.yaml
@@ -11,11 +11,11 @@ jobs:
     steps:
     - name: Checkout
       uses: actions/checkout@v3
-
-    - name: Install dependencies
-      run: |
-        pip3 install websocket-client
 
     - name: Update site
+      shell: bash
+      env:
+        CSRF_TOKEN: {{ secrets.ION_CSRF_TOKEN }}
+        SESSION_ID: {{ secrets.ION_SESSION_ID }}
       run: |
-        python3 update.py {{ secrets.ION_CSRF_TOKEN }} {{ secrets.ION_SESSION_ID }}
+        python3 update.py
diff --git a/update.py b/update.py
@@ -1,8 +1,8 @@
 from websocket import WebSocket
-import sys
+import os
 
 conn = WebSocket()
-conn.connect("wss://director.tjhsst.edu/sites/399/terminal/?", cookie=f"csrftoken={sys.argv[1]};sessionid={sys.argv[2]}")
+conn.connect("wss://director.tjhsst.edu/sites/399/terminal/?", cookie=f"csrftoken={os.environ["CSRF_TOKEN"]};sessionid={os.environ["SESSION_ID"]}")
 if 'true' not in conn.recv():
     print("Could not connect")
     exit(1)