mirror of
https://github.com/JoseDeFreitas/awesome-youtubers.git
synced 2024-07-29 14:21:52 +00:00
103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
import json
|
|
from datetime import datetime
|
|
from flask import Flask, jsonify, make_response, request
|
|
from flask_limiter import Limiter
|
|
from flask_limiter.util import get_remote_address
|
|
|
|
app = Flask(__name__)
|
|
|
|
limiter = Limiter(
|
|
app,
|
|
key_func=get_remote_address,
|
|
default_limits=["1200 per day", "50 per hour"]
|
|
)
|
|
|
|
with open("data.json", "r", encoding="utf8") as read_data:
|
|
channels = json.load(read_data)
|
|
|
|
|
|
@app.route("/")
|
|
def main():
|
|
return "Awesome YouTubers voting system website."
|
|
|
|
|
|
@app.route("/channels/all")
|
|
def list_channels():
|
|
""" Lists all channels in the database. """
|
|
|
|
return jsonify(channels)
|
|
|
|
|
|
@app.route("/channels/<channel>")
|
|
def get_channel(channel):
|
|
"""
|
|
If no query specified, prints the name of the
|
|
YouTube channel typed. When a query with the
|
|
name "vote" is given, adds or substracts 1
|
|
from the specified YouTube score.
|
|
"""
|
|
|
|
if "vote" in request.args:
|
|
vote = str(request.args["vote"])
|
|
|
|
if channel in channels:
|
|
# Adds/substracts 1 from the channel.
|
|
if vote == "upvote":
|
|
channels[channel] += 1
|
|
elif vote == "downvote":
|
|
channels[channel] -= 1
|
|
else:
|
|
return "Vote word not recognised."
|
|
|
|
# Write to database file.
|
|
with open("data.json", "w", encoding="utf8") as write_data:
|
|
json.dump(channels, write_data, indent=4)
|
|
|
|
# Write to log file.
|
|
with open("log.txt", "a") as append:
|
|
today = datetime.today().strftime('%Y-%m-%d-%H:%M')
|
|
append.write(f"\n{vote.title()}d {channel} on {today}")
|
|
|
|
return f"You {vote}d successfully the channel {channel}."
|
|
else:
|
|
return "Channel not found on the list."
|
|
else:
|
|
if channel in channels:
|
|
return "Channel: " + channel
|
|
else:
|
|
return "Channel not found on the list."
|
|
|
|
|
|
@app.route("/channels/<channel>/image.svg")
|
|
def img_channel(channel):
|
|
""" Returns the YouTube score in a svg image. """
|
|
|
|
svg_image = f"""
|
|
<svg xmlns="http://www.w3.org/2000/svg" x="0px" y="0px"
|
|
width="52px" height="22px" viewBox="0 0 52 22" fill="none">
|
|
<style>
|
|
.text {{
|
|
font-family: "Segoe UI", Ubuntu, Sans-Serif;
|
|
font-weight: bold;
|
|
}}
|
|
</style>
|
|
<rect x="0.5" y="0.5" height="99%" width="51" fill="none"/>
|
|
<g>
|
|
<text x="5" y="15" fill="#00b4f0" class="text">
|
|
{channels[channel]}
|
|
</text>
|
|
</g>
|
|
</svg>
|
|
"""
|
|
|
|
if channel in channels:
|
|
response = make_response(svg_image)
|
|
response.headers.set('Content-Type', 'image/svg+xml')
|
|
return response
|
|
else:
|
|
return "Channel not found on the list"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run()
|