db update + teams

main
Valentin 2023-11-16 15:45:38 +01:00
parent 245ffb5a7a
commit c7601238ec
1 changed files with 17 additions and 5 deletions

22
app.py
View File

@ -34,16 +34,29 @@ class User(db.Model):
class Team(db.Model): class Team(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(120), unique=True) idTeam = db.Column(db.Integer, unique=True)
idUser = db.Column(db.Integer, db.ForeignKey('user.id')) idUser = db.Column(db.Integer, db.ForeignKey('user.id'))
def addTeam(idUser, idTeam):
team = Team(idTeam=idTeam, idUser=idUser)
db.session.add(team)
db.session.commit()
def getTeamName(idTeam):
def getTeams(idUser):
teams = Team.query.filter_by(idUser=idUser).all()
return teams
@app.route('/') @app.route('/')
def index(): def index():
if 'google_token' in session: if 'google_token' in session:
me = google.get('userinfo') me = google.get('userinfo')
user = User.query.filter_by(google_id=me.data['id']).first() user = User.query.filter_by(email=me.data['email']).first()
if user is None: if user is None:
user = User(google_id=me.data['id'], email=me.data['email']) user = User(email=me.data['email'])
db.session.add(user) db.session.add(user)
db.session.commit() db.session.commit()
return 'Logged in as: ' + me.data['email'] return 'Logged in as: ' + me.data['email']
@ -56,7 +69,7 @@ def login():
@app.route('/logout') @app.route('/logout')
def logout(): def logout():
session.pop('google_token', None) session.pop('google_token', None)
return 'Logged out' return redirect("/", code=302)
@app.route('/login/authorized') @app.route('/login/authorized')
def authorized(): def authorized():
@ -67,7 +80,6 @@ def authorized():
request.args['error_description'] request.args['error_description']
) )
session['google_token'] = (response['access_token'], '') session['google_token'] = (response['access_token'], '')
me = google.get('userinfo')
return redirect("/", code=302) return redirect("/", code=302)
@google.tokengetter @google.tokengetter