-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathapp.py
109 lines (90 loc) · 2.98 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
from flask import Flask, render_template, request, redirect
from flask_sqlalchemy import SQLAlchemy
from datetime import datetime
global_users = {}
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/user_database1.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
db = SQLAlchemy(app)
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), unique=True)
full_name = db.Column(db.String)
calendar = db.Column(db.String)
groups = db.relationship('Group', backref='user', lazy='dynamic')
def __init__(self, name, calendar=None, full_name=None):
self.name = name
self.full_name = full_name
self.calendar = calendar
class Group(db.Model):
id = db.Column(db.Integer, primary_key=True)
group_code = db.Column(db.String, unique=False)
person_id = db.Column(db.Integer, db.ForeignKey('user.id'))
def __init__(self, group_code, person_id):
self.group_code = group_code
self.person_id = person_id
def generate_code():
key = str(hash(str(datetime.now())))
key = key[0:6]
return str(key)
@app.route('/')
def home():
return render_template('index.html')
@app.route('/', methods=['POST'])
def check_username():
username_exists = False
username = request.form['username']
db_user = User.query.filter_by(name=username).first()
if db_user is not None:
username_exists = True
url = '/user/' + username
return redirect(url)
else:
username_exists = False
return render_template('index.html', username_exists=username_exists)
@app.route('/new_user/')
def new_user():
username = generate_code()
user = User(username)
db.session.add(user)
db.session.commit()
global_users[username] = []
url = '/user/' + username
return redirect(url)
@app.route('/user/<username>')
def user(username):
db_user = User.query.filter_by(name=username).first()
events = global_users[username]
return render_template('user.html', user=db_user, events=events)
@app.route('/user/<username>/create_event/<start>/<end>')
def create_event(username, start, end):
event = [start, end]
global_users[username].append(event)
url = '/user/' + username
return redirect(url)
@app.route('/new_group/<username>')
def new_group(username):
group_code = generate_code()
person_id = User.query.filter_by(name=username).first().id
group = Group(group_code, person_id)
db.session.add(group)
db.session.commit()
url = '/group/' + group_code
return redirect(url)
@app.route('/group/<group_code>')
def group(group_code):
users = Group.query.filter_by(group_code=group_code).all()
group_code = group_code
return render_template('group.html', users=users, group_code=group_code)
@app.route('/group/<group_code>/add_users', methods=['POST'])
def add_user(group_code):
username = request.form['username']
person_id = User.query.filter_by(name=username).first().id
new_user = Group(group_code, person_id)
db.session.add(new_user)
db.session.commit()
url = '/group/' + group_code
return redirect(url)
if __name__ == '__main__':
db.create_all()
app.run(debug=True)