Spaces:
Sleeping
Sleeping
Upload Flask app
Browse files- Dockerfile +26 -0
- app.py +165 -0
- models.py +10 -0
- requirements.txt +7 -0
Dockerfile
ADDED
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Sử dụng một image chính thức của Python 3
|
2 |
+
FROM python:3.9-slim
|
3 |
+
|
4 |
+
# Thiết lập biến môi trường để Python không tạo các file pyc
|
5 |
+
ENV PYTHONDONTWRITEBYTECODE=1
|
6 |
+
|
7 |
+
# Thiết lập biến môi trường để Python không buffer output
|
8 |
+
ENV PYTHONUNBUFFERED=1
|
9 |
+
|
10 |
+
# Tạo và đặt thư mục làm việc cho ứng dụng Flask
|
11 |
+
WORKDIR /app
|
12 |
+
|
13 |
+
# Sao chép file requirements.txt vào container
|
14 |
+
COPY requirements.txt /app/
|
15 |
+
|
16 |
+
# Cài đặt các dependencies từ requirements.txt
|
17 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
18 |
+
|
19 |
+
# Sao chép toàn bộ mã nguồn ứng dụng vào container
|
20 |
+
COPY . /app/
|
21 |
+
|
22 |
+
# Expose port 5000 for Flask
|
23 |
+
EXPOSE 5000
|
24 |
+
|
25 |
+
# Chạy ứng dụng Flask bằng Gunicorn
|
26 |
+
CMD ["gunicorn", "-w", "4", "-b", "0.0.0.0:5000", "app:app"]
|
app.py
ADDED
@@ -0,0 +1,165 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from flask import Flask, request, jsonify, render_template
|
2 |
+
from models import db, WifiSignal
|
3 |
+
from datetime import datetime, timedelta
|
4 |
+
import random
|
5 |
+
import folium
|
6 |
+
from folium.plugins import MarkerCluster
|
7 |
+
from flasgger import Swagger
|
8 |
+
|
9 |
+
app = Flask(__name__)
|
10 |
+
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
|
11 |
+
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
12 |
+
|
13 |
+
# Khởi tạo Swagger
|
14 |
+
swagger = Swagger(app)
|
15 |
+
|
16 |
+
db.init_app(app)
|
17 |
+
|
18 |
+
@app.route('/api/generate-data', methods=['POST'])
|
19 |
+
def generate_data():
|
20 |
+
"""
|
21 |
+
Generate demo data.
|
22 |
+
---
|
23 |
+
tags:
|
24 |
+
- Data Management
|
25 |
+
responses:
|
26 |
+
201:
|
27 |
+
description: Demo data generated successfully
|
28 |
+
"""
|
29 |
+
base_latitude = 35.6837
|
30 |
+
base_longitude = 139.6805
|
31 |
+
start_date = datetime(2024, 8, 1)
|
32 |
+
end_date = datetime(2024, 8, 7)
|
33 |
+
delta = end_date - start_date
|
34 |
+
|
35 |
+
for _ in range(100):
|
36 |
+
random_days = random.randint(0, delta.days)
|
37 |
+
random_seconds = random.randint(0, 86400)
|
38 |
+
random_time = start_date + timedelta(days=random_days, seconds=random_seconds)
|
39 |
+
|
40 |
+
random_latitude = base_latitude + random.uniform(-0.01, 0.01)
|
41 |
+
random_longitude = base_longitude + random.uniform(-0.01, 0.01)
|
42 |
+
random_signal_strength = random.randint(10, 50)
|
43 |
+
|
44 |
+
wifi_signal = WifiSignal(
|
45 |
+
latitude=random_latitude,
|
46 |
+
longitude=random_longitude,
|
47 |
+
timestamp=random_time,
|
48 |
+
signal_strength=random_signal_strength
|
49 |
+
)
|
50 |
+
|
51 |
+
db.session.add(wifi_signal)
|
52 |
+
|
53 |
+
db.session.commit()
|
54 |
+
return jsonify({"message": "Demo data generated successfully"}), 201
|
55 |
+
|
56 |
+
@app.route('/api/delete-data', methods=['DELETE'])
|
57 |
+
def delete_data():
|
58 |
+
"""
|
59 |
+
Delete all data in the database.
|
60 |
+
---
|
61 |
+
tags:
|
62 |
+
- Data Management
|
63 |
+
responses:
|
64 |
+
200:
|
65 |
+
description: All data deleted successfully
|
66 |
+
500:
|
67 |
+
description: An error occurred
|
68 |
+
"""
|
69 |
+
try:
|
70 |
+
num_rows_deleted = db.session.query(WifiSignal).delete()
|
71 |
+
db.session.commit()
|
72 |
+
return jsonify({"message": f"Deleted {num_rows_deleted} rows from the database."}), 200
|
73 |
+
except Exception as e:
|
74 |
+
db.session.rollback()
|
75 |
+
return jsonify({"message": f"An error occurred: {str(e)}"}), 500
|
76 |
+
|
77 |
+
@app.route('/api/upload', methods=['POST'])
|
78 |
+
def upload_data():
|
79 |
+
"""
|
80 |
+
Upload WiFi signal data.
|
81 |
+
---
|
82 |
+
tags:
|
83 |
+
- Data Management
|
84 |
+
parameters:
|
85 |
+
- name: body
|
86 |
+
in: body
|
87 |
+
required: true
|
88 |
+
schema:
|
89 |
+
type: object
|
90 |
+
properties:
|
91 |
+
latitude:
|
92 |
+
type: number
|
93 |
+
example: 35.6895
|
94 |
+
longitude:
|
95 |
+
type: number
|
96 |
+
example: 139.6917
|
97 |
+
timestamp:
|
98 |
+
type: string
|
99 |
+
example: "2024-08-29 14:30:00"
|
100 |
+
signal_strength:
|
101 |
+
type: integer
|
102 |
+
example: 42
|
103 |
+
responses:
|
104 |
+
201:
|
105 |
+
description: Data uploaded successfully
|
106 |
+
400:
|
107 |
+
description: Invalid input
|
108 |
+
"""
|
109 |
+
data = request.json
|
110 |
+
latitude = data.get('latitude')
|
111 |
+
longitude = data.get('longitude')
|
112 |
+
timestamp = datetime.strptime(data.get('timestamp'), '%Y-%m-%d %H:%M:%S')
|
113 |
+
signal_strength = data.get('signal_strength')
|
114 |
+
|
115 |
+
wifi_signal = WifiSignal(latitude=latitude, longitude=longitude, timestamp=timestamp, signal_strength=signal_strength)
|
116 |
+
db.session.add(wifi_signal)
|
117 |
+
db.session.commit()
|
118 |
+
|
119 |
+
return jsonify({"message": "Data uploaded successfully"}), 201
|
120 |
+
|
121 |
+
@app.route('/', methods=['GET'])
|
122 |
+
def show_map():
|
123 |
+
start_date = request.args.get('start_date')
|
124 |
+
end_date = request.args.get('end_date')
|
125 |
+
|
126 |
+
query = WifiSignal.query
|
127 |
+
|
128 |
+
if start_date and end_date:
|
129 |
+
start_datetime = datetime.strptime(start_date, '%Y-%m-%d')
|
130 |
+
end_datetime = datetime.strptime(end_date, '%Y-%m-%d')
|
131 |
+
query = query.filter(WifiSignal.timestamp >= start_datetime, WifiSignal.timestamp <= end_datetime)
|
132 |
+
|
133 |
+
signals = query.all()
|
134 |
+
signal_data = [(signal.latitude, signal.longitude, signal.signal_strength) for signal in signals]
|
135 |
+
|
136 |
+
if signal_data:
|
137 |
+
min_signal = min(s[2] for s in signal_data)
|
138 |
+
max_signal = max(s[2] for s in signal_data)
|
139 |
+
else:
|
140 |
+
min_signal = max_signal = 1
|
141 |
+
|
142 |
+
m = folium.Map(location=[35.6837, 139.6805], zoom_start=12)
|
143 |
+
marker_cluster = MarkerCluster().add_to(m)
|
144 |
+
|
145 |
+
for lat, lon, signal_strength in signal_data:
|
146 |
+
if max_signal - min_signal != 0:
|
147 |
+
radius = 5 + 10 * ((signal_strength - min_signal) / (max_signal - min_signal))
|
148 |
+
else:
|
149 |
+
radius = 15
|
150 |
+
folium.CircleMarker(
|
151 |
+
location=[lat, lon],
|
152 |
+
radius=radius,
|
153 |
+
popup=f'Signal Strength: {signal_strength}',
|
154 |
+
color='blue',
|
155 |
+
fill=True,
|
156 |
+
fill_opacity=0.6
|
157 |
+
).add_to(marker_cluster)
|
158 |
+
|
159 |
+
map_html = m._repr_html_()
|
160 |
+
return render_template('map.html', map_html=map_html)
|
161 |
+
|
162 |
+
if __name__ == '__main__':
|
163 |
+
with app.app_context():
|
164 |
+
db.create_all()
|
165 |
+
app.run(debug=True)
|
models.py
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from flask_sqlalchemy import SQLAlchemy
|
2 |
+
|
3 |
+
db = SQLAlchemy()
|
4 |
+
|
5 |
+
class WifiSignal(db.Model):
|
6 |
+
id = db.Column(db.Integer, primary_key=True)
|
7 |
+
latitude = db.Column(db.Float, nullable=False)
|
8 |
+
longitude = db.Column(db.Float, nullable=False)
|
9 |
+
timestamp = db.Column(db.DateTime, nullable=False)
|
10 |
+
signal_strength = db.Column(db.Integer, nullable=False)
|
requirements.txt
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
Flask==2.2.5
|
2 |
+
Flask-SQLAlchemy==3.0.3
|
3 |
+
Flasgger==0.9.5
|
4 |
+
folium==0.13.0
|
5 |
+
gunicorn==20.1.0
|
6 |
+
|
7 |
+
|