mirror of https://github.com/itsmrval/somycal
db update + teams
parent
245ffb5a7a
commit
c7601238ec
22
app.py
22
app.py
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue