import random import operator import json from flask import Flask, Response, redirect, url_for, request, session, abort, render_template from flask_login import LoginManager, UserMixin, login_required, login_user, logout_user, current_user app = Flask(__name__) #config app.config.update( DEBUG = True, SECRET_KEY = 'secret_xxx' ) configFileName = "kukconfig.json" try: with open (configFileName, 'r') as configFile: config = json.load(configFile) except IOError: print "Config file not found! Loading defaults" config = {} config['ip'] = '0.0.0.0' config['port'] = 5000 config['debug'] = True config['mealHistoryFile'] = "MealHistory.json" config['peopleFile'] = "People.json" people = {} class Person(UserMixin): global people def __init__(self, name): self.name = name self.id = name self.kukPoints = 0 self.password = '' people[name] = self @classmethod def fromFile(self, name, kukPoints, password): person = Person(name) person.kukPoints = kukPoints person.password = password @staticmethod def get(name): if name not in people: people[name] = Person(name) return people[name] @staticmethod def leaderboard(): leaderboard = [] for name,person in people.iteritems(): entry = {} entry['name'] = name entry['score'] = person.kukPoints leaderboard.append(entry) leaderboard.sort(key=lambda tup : tup['score'], reverse = True) return leaderboard def serialized(self): return self.__dict__ mealHistory = {} class Meal(object): global mealHistory def __init__(self): self.mid = '' self.kuk = '' self.eaters = [] self.date = '' self.flavorText = '' self.accounted = "" @classmethod def fromFile(self, mid, kuk, eaters, date, flavorText, accounted): meal = Meal() meal.mid = mid meal.kuk = kuk meal.eaters = eaters meal.date = date meal.flavorText = flavorText meal.accounted = accounted mealHistory[meal.mid] = meal return meal @classmethod def new(self, date): meal = Meal() meal.mid = self.getNextMID() meal.date = date meal.accounted = "no" mealHistory[meal.mid] = meal return meal def account(self): if self.accounted != "no": return "ERROR! this meal has already been accounted for or is broken!" self.accounted = "yes" eaterCount = len(self.eaters) points = config['points'][eaterCount] print "kuk {} clears {} points".format(self.kuk, points) Person.get(self.kuk).kukPoints -= points for eater in self.eaters: Person.get(eater).kukPoints += points / eaterCount print "+{} gets {} points ".format(eater, points / eaterCount) @staticmethod def getCurrent(): return mealHistory[max(mealHistory.keys())] @staticmethod def getNextMID(): if len(mealHistory) == 0: return 1 return max(mealHistory.keys())+1 def serialized(self): return self.__dict__ def save(): with open(config['mealHistoryFile'], 'w') as outfile: mealListJson = [] for mid,meal in mealHistory.iteritems(): mealListJson.append(meal.serialized()) json.dump(mealListJson, outfile) with open(config['peopleFile'], 'w') as outfile: peopleJson = [] for name,person in people.iteritems(): peopleJson.append(person.serialized()) json.dump(peopleJson, outfile) def load(): mealHistory = {} people = {} try: with open(config['mealHistoryFile'], 'r') as infile: fileData = json.load(infile) except: print "no meal history file!" return for meal in fileData: Meal.fromFile(meal['mid'], meal['kuk'], meal['eaters'], meal['date'], meal['flavorText'], meal['accounted']) try: with open(config['peopleFile'], 'r') as infile: fileData = json.load(infile) except: print "no people file!" return for person in fileData: Person.fromFile(person['name'], person['kukPoints'], person['password']) load() # flask-login login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = "login" @app.route("/") @login_required def index(): meal = Meal.getCurrent() cook = {} cook['name'] = meal.kuk cook['confirmed'] = 'yes' return render_template('index.html', leaderboard=Person.leaderboard(), cook=cook, eaters=meal.eaters) # somewhere to login @app.route("/login", methods=["GET", "POST"]) def login(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] print len(people) if Person.get(username).password == password: login_user(Person.get(username)) return redirect(request.args.get("next")) return abort(401) else: return Response('''

''') # adding to meal @app.route("/addme") @login_required def addme(): meal = Meal.getCurrent() if current_user.name not in meal.eaters: meal.eaters.append(current_user.name) return Response('

Have a nice meal!

') else: return Response('

only one meal per person!

') #anti-adding to meal @app.route("/removeme") @login_required def removeme(): meal = Meal.getCurrent() if current_user.name in meal.eaters: meal.eaters.remove(current_user.name) return Response('

no food for you!

') else: return Response('

insert joke here

') # somewhere to logout @app.route("/logout") @login_required def logout(): logout_user() return Response('

Logged out

') # handle login failed @app.errorhandler(401) def page_not_found(e): return Response('

Login failed

') # callback to reload the user object @login_manager.user_loader def load_user(userid): return Person.get(userid)