HeadlineClass.py

#!/usr/bin/env python3
from flask import Flask, render_template, request
from datetime import datetime, timedelta, date
import random
import json

app = Flask(__name__)
highscores = {}
class HeadlineLoader:
    def __init__(self, json_file):
        self.headlines = []
        self.last_headline = None

        with open(json_file) as f:
            i = 0
            for line in f:
                try:
                    data = json.loads(line)
                    if data["category"] in ["POLITICS", "QUEER VOICES", "BLACK VOICES", "TECH", "BUSINESS"] and len(data["headline"]) >80:
                        headline = {
                            "id": i,
                            "link": data["link"],
                            "headline": data["headline"],
                            "date": datetime.strptime(data["date"], "%Y-%m-%d").date(),
                            "highscore": 99999
                        }

                        self.headlines.append(headline)
                        i+=1
                except ValueError:
                    pass
                    
    def get_random_headline(self):
        if self.headlines:
            i = random.randrange(0,len(self.headlines))
            self.headlines[i]['id'] = i
            self.last_headline = self.headlines[i]
            return self.last_headline
        else:
            return None
        
    def get_random_headlines(self, num_headlines):
        if len(self.headlines) < num_headlines:
            return None
        random_headlines = random.sample(self.headlines, num_headlines)
        return random_headlines

    def get_last_headline(self):
        if self.last_headline == None:
            return None
        else:
            return self.last_headline
    def get_headline_by_headline(self, headline):
        for item in self.headlines:
            if item["headline"] == headline:
                return item
            return None
        
    def get_headline_by_id(self, id):
        for item in self.headlines:
            #print("should match at some point", item.get('id'), id)
            if item.get('id') == id:
                print("found Item")
                return item
        print("Failed to find headline")
        return self.last_headline
json_file = "News_Category_Dataset_v3.json"
headline_loader = HeadlineLoader(json_file)

@app.route('/')
def site_index():
    return render_template('site_index.html')


@app.route('/game')
def game():
    random_headline = headline_loader.get_random_headline()
    return render_template('index.html', headline=random_headline)

@app.route('/guess', methods=['POST'])
def guess():
    guess_date = request.form.get('date')
    headline = request.form.get('headline')
    actual_date = request.form.get('actual_date')
    link = request.form.get('link')
    print("AAAAAAAAAA", guess_date, headline, actual_date)
    try:
        actual_date = datetime.strptime(actual_date, "%Y-%m-%d").date()
        guess_date = datetime.strptime(guess_date, "%Y-%m-%d").date()
    except:
        guess_date = date(2002, 12, 31)
        actual_date = date(2022, 12, 31)
    
    days_off = abs((guess_date - actual_date).days)
    if link in highscores.keys():
        highscore = highscores[link]
    else:
        highscore = 9999999
    if days_off < highscore:
        score_text = "You set a NEW high score of: " + str(days_off) 
        prior_score_text = '''The prior high score was: '''+ str(highscore)
        highscores[link] = days_off
    else:
        score_text = "The high score is: " + str(highscore)
        prior_score_text = " "
    return render_template('result.html', headline=headline, actual_date=actual_date, guess_date=guess_date, days_off=days_off, link=link, score_text=score_text, prior_score_text=prior_score_text)


@app.route('/order-game', methods=['GET'])
def order_game():
    num_headlines = 2
    random_headlines = headline_loader.get_random_headlines(num_headlines)
    random.shuffle(random_headlines)
    print(random_headlines)
    return render_template('order_game.html', headlines=random_headlines, num_headlines=num_headlines, url="/order-game")

@app.route('/medium-difficulty', methods=['GET'])
def order_game_medium():
    num_headlines = 3
    random_headlines = headline_loader.get_random_headlines(num_headlines)
    random.shuffle(random_headlines)
    print(random_headlines)
    return render_template('order_game.html', headlines=random_headlines, num_headlines=num_headlines,url="/medium-difficulty")

@app.route('/hard-difficulty', methods=['GET'])
def order_game_hard():
    num_headlines = 5
    random_headlines = headline_loader.get_random_headlines(num_headlines)
    random.shuffle(random_headlines)
    print(random_headlines)
    return render_template('order_game.html', headlines=random_headlines, num_headlines=num_headlines, url="/hard-difficulty")

@app.route('/ultra-difficulty', methods=['GET'])
def order_game_ultra():
    num_headlines = 7
    random_headlines = headline_loader.get_random_headlines(num_headlines)
    random.shuffle(random_headlines)
    print(random_headlines)
    return render_template('order_game.html', headlines=random_headlines, num_headlines=num_headlines, url="/ultra-difficulty")

@app.route('/order-game-guess', methods=['POST'])
def order_game_guess():
    headlineIDs = request.get_json()
    print(headlineIDs)
    headlinesUserSort = []
    for id in headlineIDs:
        headlinesUserSort.append(headline_loader.get_headline_by_id(id[0]))
    for i in range(len(headlineIDs)-1):
        if headlinesUserSort[i]["date"] < headlinesUserSort[i+1]["date"]:
            print("comp", headlinesUserSort[i]["date"], headlinesUserSort[i+1]["date"])
            pass
        else:
            return {"success":False}
    return {"success":True}


if __name__ == '__main__':
    app.run(host='209.38.248.210', port=80)