|
|
import asyncio
|
|
|
import json
|
|
|
import os
|
|
|
import re
|
|
|
import sys
|
|
|
from datetime import datetime
|
|
|
from typing import cast
|
|
|
import threading
|
|
|
|
|
|
import tornado.web
|
|
|
import tornado.websocket
|
|
|
from tornado import ioloop
|
|
|
|
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
|
from prophet_model import ProphetModel
|
|
|
|
|
|
ISO_PATTERN = r'^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$'
|
|
|
DATASET = os.getenv("DATASET", "data/traffic_data.csv")
|
|
|
|
|
|
class MainHandler(tornado.web.RequestHandler):
|
|
|
"""
|
|
|
Handler to render the index page
|
|
|
"""
|
|
|
def get(self):
|
|
|
self.render("index.html")
|
|
|
|
|
|
class PredictionHandler(tornado.web.RequestHandler):
|
|
|
"""
|
|
|
Handler to process the request for predicting traffic data
|
|
|
"""
|
|
|
def initialize(self):
|
|
|
"""
|
|
|
Initializes and trains a model
|
|
|
"""
|
|
|
self.model = ProphetModel()
|
|
|
self.model.train_model(DATASET)
|
|
|
|
|
|
def post(self):
|
|
|
"""
|
|
|
POST method to process the request for traffic data generation
|
|
|
"""
|
|
|
try:
|
|
|
print(self.request.body)
|
|
|
run_parameters = json.loads(self.request.body)
|
|
|
|
|
|
if run_parameters is None:
|
|
|
self.write("Invalid parameters: 400")
|
|
|
self.set_status(400)
|
|
|
return
|
|
|
|
|
|
predict_time = cast(str, run_parameters.get("predict_time"))
|
|
|
|
|
|
if re.match(ISO_PATTERN, predict_time):
|
|
|
|
|
|
predict_time = datetime.strptime(predict_time, "%Y-%m-%d %H:%M:%S")
|
|
|
else:
|
|
|
self.write("Please provide date in the format YYYY-MM-DD HH:mm:ss")
|
|
|
self.set_status(400)
|
|
|
return
|
|
|
|
|
|
dataset = self.model.predict(predict_time)
|
|
|
self.write(json.dumps(dataset))
|
|
|
|
|
|
except json.JSONDecodeError:
|
|
|
self.write("Invalid JSON payload: 400")
|
|
|
self.set_status(400)
|
|
|
return
|
|
|
except Exception as error_message:
|
|
|
self.set_status(500)
|
|
|
self.write(f"Error in data generation: {str(error_message)}")
|
|
|
return
|
|
|
|
|
|
def make_app():
|
|
|
"""
|
|
|
Create the tornado application
|
|
|
"""
|
|
|
return tornado.web.Application([
|
|
|
(r"/", MainHandler),
|
|
|
(r"/predict", PredictionHandler),
|
|
|
])
|
|
|
|
|
|
def start():
|
|
|
"""
|
|
|
Start the server
|
|
|
"""
|
|
|
def run_server():
|
|
|
asyncio.set_event_loop(asyncio.new_event_loop())
|
|
|
try:
|
|
|
app = make_app()
|
|
|
app.listen(8889)
|
|
|
print("Server is running on port 8889")
|
|
|
ioloop.IOLoop.current().start()
|
|
|
except Exception as error_message:
|
|
|
print("An error occurred while starting the server:", error_message)
|
|
|
|
|
|
server_thread = threading.Thread(target=run_server)
|
|
|
server_thread.start()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
start()
|
|
|
|
