vumichien commited on
Commit
737ad85
·
1 Parent(s): 13f4fea

Upload Flask app

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app.py +5 -3
Dockerfile CHANGED
@@ -37,4 +37,4 @@ USER appuser
37
  EXPOSE 5000
38
 
39
  # Chạy ứng dụng FastAPI bằng Uvicorn
40
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"]
 
37
  EXPOSE 5000
38
 
39
  # Chạy ứng dụng FastAPI bằng Uvicorn
40
+ CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"]
app.py CHANGED
@@ -9,10 +9,10 @@ from datetime import datetime, timedelta
9
  import random
10
  import folium
11
  from folium.plugins import MarkerCluster
 
12
 
13
  # Sử dụng thư mục /tmp để lưu trữ cơ sở dữ liệu SQLite
14
- DATABASE_URL = "sqlite:///:memory:"
15
- # DATABASE_URL = "sqlite:///tmp/database.db"
16
 
17
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
18
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -27,7 +27,9 @@ class WifiSignal(Base):
27
  timestamp = Column(DateTime, index=True)
28
  signal_strength = Column(Integer)
29
 
30
- Base.metadata.create_all(bind=engine)
 
 
31
 
32
  app = FastAPI()
33
  templates = Jinja2Templates(directory="templates")
 
9
  import random
10
  import folium
11
  from folium.plugins import MarkerCluster
12
+ import os
13
 
14
  # Sử dụng thư mục /tmp để lưu trữ cơ sở dữ liệu SQLite
15
+ DATABASE_URL = "sqlite:///./database.db"
 
16
 
17
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
18
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
27
  timestamp = Column(DateTime, index=True)
28
  signal_strength = Column(Integer)
29
 
30
+ # Tạo tất cả các bảng
31
+ if not os.path.exists('/app/database.db'):
32
+ Base.metadata.create_all(bind=engine)
33
 
34
  app = FastAPI()
35
  templates = Jinja2Templates(directory="templates")