home · contact · privacy
Add rump HTTP server/handler testing.
[plomtask] / run.py
diff --git a/run.py b/run.py
index ef6aeab6b7fd56a93ed3c9e76a7b5bbfe532b68c..31d11cee0534b5ae68c3bf84692f61b87b9e7481 100755 (executable)
--- a/run.py
+++ b/run.py
@@ -6,17 +6,16 @@ from plomtask.misc import HandledException
 from plomtask.http import TaskHandler, TaskServer
 from plomtask.db import DatabaseFile
 
-PATH_DB = environ.get('PATH_DB')
+PLOMTASK_DB_PATH = environ.get('PLOMTASK_DB_PATH')
 HTTP_PORT = 8082
-TEMPLATES_DIR = 'templates'
 DB_CREATION_ASK = 'Database file not found. Create? Y/n\n'
 
 
 if __name__ == '__main__':
     try:
-        if not PATH_DB:
-            raise HandledException('PATH_DB not set.')
-        db_file = DatabaseFile(PATH_DB)
+        if not PLOMTASK_DB_PATH:
+            raise HandledException('PLOMTASK_DB_PATH not set.')
+        db_file = DatabaseFile(PLOMTASK_DB_PATH)
         if not db_file.exists:
             legal_yesses = {'y', 'yes', 'yes.', 'yes!'}
             reply = input(DB_CREATION_ASK)
@@ -25,8 +24,7 @@ if __name__ == '__main__':
             else:
                 print('Not recognizing reply as "yes".')
                 raise HandledException('Cannot run without database.')
-        server = TaskServer(TEMPLATES_DIR, db_file,
-                            ('localhost', HTTP_PORT), TaskHandler)
+        server = TaskServer(db_file, ('localhost', HTTP_PORT), TaskHandler)
         print(f'running at port {HTTP_PORT}')
         try:
             server.serve_forever()