From 910e9293949c807c207d15676b3ac44f38ea4c33 Mon Sep 17 00:00:00 2001 From: livindra Date: Tue, 28 Apr 2026 21:04:01 +0700 Subject: [PATCH] db fiks --- app.py | 2 +- setup_db.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/app.py b/app.py index 155c840..e379645 100644 --- a/app.py +++ b/app.py @@ -911,6 +911,6 @@ def export_excel(): if __name__ == '__main__': - port = os.environ.get('PORT', 5000) + port = int(os.environ.get('PORT', 5000)) debug = os.environ.get('FLASK_DEBUG', '0') == '1' app.run(host='0.0.0.0', port=port, debug=debug) \ No newline at end of file diff --git a/setup_db.py b/setup_db.py index ff17aed..d6d2b42 100644 --- a/setup_db.py +++ b/setup_db.py @@ -13,12 +13,17 @@ from pymysql import MySQLError load_dotenv() DB_HOST = os.environ.get('DB_HOST', 'localhost') + DB_PORT = os.environ.get('DB_PORT', '3306') +try: + DB_PORT = int(DB_PORT) +except: + DB_PORT = 3306 + DB_USER = os.environ.get('DB_USER', 'root') DB_PASSWORD = os.environ.get('DB_PASSWORD', '') DB_NAME = os.environ.get('DB_NAME', 'deteksi_pmk') - def create_database(): """Create database if not exists""" try: