Merge pull request #54 from edoardottt/devel

Devel - Add api endpoints

Author: edoardottt

db.py

@@ -94,6 +94,23 @@ def user_stats(conn, username):
     return values
 
 
+def user_date_stats(conn, username, date):
+    """
+    This function retrieves all the records for a user for a specific day.
+    """
+    sql = """ SELECT * FROM statistics WHERE username = ? AND date = ?"""
+    cur = conn.cursor()
+    cur.execute(
+        sql,
+        (
+            username,
+            date,
+        ),
+    )
+    values = cur.fetchall()
+    return values
+
+
 def all_stats(conn):
     """
     This function retrieves all the records.

server.py

@@ -17,6 +17,7 @@
 from flask.templating import render_template
 import datetime
 import db
+from dateutil.parser import parse
 
 app = Flask(__name__, template_folder="templates")
 app.config["SECRET_KEY"] = "ILVYilvbthLQETHeteggrgwi2r389"
@@ -123,6 +124,9 @@ def user_dashboard():
 
 @app.route("/api/tweets/<user>")
 def api_user_tweets(user):
+    """
+    Total tweets for user <user>.
+    """
     if not user_ok(user):
         return "ERROR: Invalid username."
 
@@ -140,6 +144,9 @@ def api_user_tweets(user):
 
 @app.route("/api/likes/<user>")
 def api_user_likes(user):
+    """
+    Total likes for user <user>.
+    """
     if not user_ok(user):
         return "ERROR: Invalid username."
 
@@ -157,6 +164,9 @@ def api_user_likes(user):
 
 @app.route("/api/retweets/<user>")
 def api_user_retweets(user):
+    """
+    Total retweets for user <user>.
+    """
     if not user_ok(user):
         return "ERROR: Invalid username."
 
@@ -174,6 +184,9 @@ def api_user_retweets(user):
 
 @app.route("/api/followers/<user>")
 def api_user_followers(user):
+    """
+    Latest follower count for user <user>.
+    """
     if not user_ok(user):
         return "ERROR: Invalid username."
 
@@ -187,5 +200,99 @@ def api_user_followers(user):
         return str(result)
 
 
+def string_to_date(date):
+    """
+    @Input: string
+    @Output: string
+    This function tries to convert a string into a date,
+    if it's not possible return an empty string.
+    """
+    try:
+        converted = parse(date)
+        return converted
+    except Exception:
+        return ""
+
+
+@app.route("/api/tweets/<user>/<date>")
+def api_user_date_tweets(user, date):
+    """
+    Tweets for user <user> in date <date>.
+    Date format: YYYY-MM-DD.
+    """
+    if not user_ok(user):
+        return "ERROR: Invalid username."
+    if string_to_date(date) == "":
+        return "ERROR: Invalid date."
+    # if checks ok ->
+    conn = db.conn_db()
+    values = db.user_date_stats(conn, user, date)
+    if values is None or len(values) == 0:
+        return "ERROR: No data for this user on this day."
+    else:
+        result = values[0][2]
+        return str(result)
+
+
+@app.route("/api/likes/<user>/<date>")
+def api_user_date_likes(user, date):
+    """
+    Likes for user <user> in date <date>.
+    Date format: YYYY-MM-DD.
+    """
+    if not user_ok(user):
+        return "ERROR: Invalid username."
+    if string_to_date(date) == "":
+        return "ERROR: Invalid date."
+    # if checks ok ->
+    conn = db.conn_db()
+    values = db.user_date_stats(conn, user, date)
+    if values is None or len(values) == 0:
+        return "ERROR: No data for this user on this day."
+    else:
+        result = values[0][3]
+        return str(result)
+
+
+@app.route("/api/retweets/<user>/<date>")
+def api_user_date_retweets(user, date):
+    """
+    Retweets for user <user> in date <date>.
+    Date format: YYYY-MM-DD.
+    """
+    if not user_ok(user):
+        return "ERROR: Invalid username."
+    if string_to_date(date) == "":
+        return "ERROR: Invalid date."
+    # if checks ok ->
+    conn = db.conn_db()
+    values = db.user_date_stats(conn, user, date)
+    if values is None or len(values) == 0:
+        return "ERROR: No data for this user on this day."
+    else:
+        result = values[0][4]
+        return str(result)
+
+
+@app.route("/api/followers/<user>/<date>")
+def api_user_date_followers(user, date):
+    """
+    Followers for user <user> in date <date>.
+    Date format: YYYY-MM-DD.
+    """
+    if not user_ok(user):
+        return "ERROR: Invalid username."
+    if string_to_date(date) == "":
+        return "ERROR: Invalid date."
+    # if checks ok ->
+    conn = db.conn_db()
+    values = db.user_date_stats(conn, user, date)
+    if values is None or len(values) == 0:
+        return "ERROR: No data for this user on this day."
+    else:
+        result = values[0][5]
+        return str(result)
+
+
 if __name__ == "__main__":
     app.run()