mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 22:43:11 +02:00
117 lines
2.8 KiB
Python
117 lines
2.8 KiB
Python
from flask import Flask, request, jsonify
|
|
from flask_caching import Cache
|
|
import psycopg2
|
|
import pandas
|
|
import json
|
|
from configparser import ConfigParser
|
|
from datetime import datetime
|
|
|
|
|
|
app = Flask(__name__)
|
|
cache = Cache(config={'CACHE_TYPE': 'SimpleCache'})
|
|
db_connector = None
|
|
conn = None
|
|
|
|
|
|
@app.route("/api/get_number_of_ratings", methods=["GET"])
|
|
@cache.cached(timeout=500)
|
|
def get_number_of_ratings():
|
|
cursor = conn.cursor()
|
|
cursor.execute("select count(*) as num_of_ratings from ratings")
|
|
res = cursor.fetchall()
|
|
|
|
cursor.close()
|
|
|
|
return jsonify(res[0]), 200
|
|
|
|
|
|
@app.route("/api/get_movie_ratings/<string:movie_id>", methods=["GET"])
|
|
@cache.cached(timeout=50)
|
|
def get_movie_ratings(movie_id):
|
|
cursor = conn.cursor()
|
|
ratings = {}
|
|
rating_values = [5, 4, 3, 2, 1]
|
|
|
|
for rating in rating_values:
|
|
cursor.execute("""
|
|
SELECT COUNT(*) as count
|
|
FROM ratings
|
|
WHERE rating = %s AND movie_ID = %s;
|
|
""", (rating, movie_id))
|
|
result = cursor.fetchone()
|
|
ratings[f'{rating}_star'] = result[0]
|
|
|
|
cursor.close()
|
|
|
|
return jsonify(ratings), 200
|
|
|
|
|
|
@app.route("/api/get_users_number", methods=["GET"])
|
|
@cache.cached(timeout=50)
|
|
def get_number_of_users():
|
|
cursor = conn.cursor()
|
|
cursor.execute("select count(*) as num_of_users from users")
|
|
res = cursor.fetchall()
|
|
|
|
cursor.close()
|
|
|
|
return jsonify(res[0]), 200
|
|
|
|
|
|
@app.route("/api/get_movie_rating_avg/<string:movie_id>", methods=["GET"])
|
|
@cache.cached(timeout=50)
|
|
def get_movie_rating_avg(movie_id):
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
SELECT AVG(rating) as avg_rating
|
|
FROM ratings
|
|
WHERE movie_ID = %s;
|
|
""", (movie_id,))
|
|
res = cursor.fetchall()
|
|
|
|
cursor.close()
|
|
|
|
return jsonify(res[0]), 200
|
|
|
|
|
|
@app.route("/api/get_user_ratings/<string:user_id>", methods=["GET"])
|
|
@cache.cached(timeout=50)
|
|
def get_user_ratings(user_id):
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
SELECT *
|
|
FROM ratings
|
|
WHERE oauth_ID = %s;
|
|
""", (user_id,))
|
|
res = cursor.fetchall()
|
|
|
|
cursor.close()
|
|
|
|
return jsonify(res), 200
|
|
|
|
|
|
if __name__ == "__main__":
|
|
config = ConfigParser()
|
|
config.read("init_scripts/constants.ini")
|
|
|
|
while True:
|
|
try:
|
|
conn = psycopg2.connect(
|
|
host=config["postgres"]["host"],
|
|
database=config["postgres"]["database"],
|
|
user=config["postgres"]["user"],
|
|
password=config["postgres"]["password"],
|
|
port=int(config["postgres"]["port"])
|
|
)
|
|
|
|
except Exception:
|
|
print("Trying to connect with database")
|
|
continue
|
|
else:
|
|
break
|
|
|
|
cache.init_app(app)
|
|
app.run(host="0.0.0.0", port=8090, debug=True)
|
|
|
|
conn.close()
|