Skip to content

Commit

Permalink
handle duplicate username/email
Browse files Browse the repository at this point in the history
  • Loading branch information
raylu committed Mar 4, 2015
1 parent 6f0b4ce commit c4c38e7
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions outlauth.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,16 @@
import eventlet
eventlet.monkey_patch()

from collections import defaultdict
import operator
import os
import string

import cleancss
import eventlet.wsgi
import flask
from flask import request, session
import eventlet.wsgi
from collections import defaultdict
import sqlalchemy.exc

import ccp_pls
import config
Expand Down Expand Up @@ -76,7 +77,14 @@ def step_2():
apikey_id=int(request.form['key_id']), apikey_vcode=request.form['vcode'],
character_id=char_id)
db.session.add(user)
db.session.commit()
try:
db.session.commit()
except sqlalchemy.exc.DBAPIError as e:
if e.orig.code == '23505': # unique_violation
flask.flash('Username or email already exists.')
return step_2()
else:
raise

session.permanent = True
session['user_id'] = user.id
Expand Down

0 comments on commit c4c38e7

Please sign in to comment.