class main_views(object):
def __init__(self, request):
self.request = request
self.session = DBSession()
self.logged_in = authenticated_userid(request)
@view_config(route_name='home', renderer='home.mak')
def home_view(self):
if not self.logged_in:
url = self.request.route_url('login')
raise HTTPFound(location=url)
user = (self.session.query(User)
.filter(User.id==self.logged_in).first())
if not user:
# Perhaps this is rude... but it only happens to deleted users
url = self.request.route_url('logout')
raise HTTPFound(location=url)
games = (self.session.query(Game)
.filter(Game.state=='ongoing')
.filter(
or_(User.id==Game.white_id,
User.id==Game.black_id))
.all())
return {'logged_in': True,
'user': user,
'games': games}
class user_views(object):
def __init__(self, request):
self.request = request
self.session = DBSession()
self.logged_in = authenticated_userid(request)
@view_config(route_name='login', renderer='login.mak',
request_method="GET")
def login_get_view(self):
if self.logged_in:
url = self.request.route_url('home')
raise HTTPFound(location=url)
return {'logged_in': False, 'message': ''}
@view_config(route_name='login', renderer='login.mak',
request_method="POST")
def login_post_view(self):
message = ''
username = self.request.POST.get('username', '')
password = self.request.POST.get('password', '')
user = (self.session.query(User)
.filter(User.username==username).first())
if user:
if user.check_password(password):
headers = remember(self.request, user.id)
url = self.request.route_url('home')
raise HTTPFound(location=url, headers=headers)
else:
message = 'Invalid password'
else:
message = 'Invalid user'
return {'logged_in': False, 'message': message}
@view_config(route_name='register', renderer='register.mak',
request_method="POST")
def register_post_vies(self):
error, message = "", ""
username = self.request.POST.get('username')
password = self.request.POST.get('password')
if not username or not re.match('^\w+$', username):
error = 'Invalid username'
else:
user = (self.session.query(User)
.filter(User.username==username).first())
if user:
error = 'Username in use'
else:
user = User.new(username, password)
self.session.add(user)
self.session.commit()
message = 'User "%s" successfully created' % username
return dict(error=error,message=message)
def __init__(self, request):
self.request = request
self.session = DBSession()
class api_views(object):
def __init__(self, request):
self.request = request
self.session = DBSession()
@view_config(route_name="users", renderer='json',
permission="view", request_method="GET")
def list_users_view(self):
return [u.__json__() for u in self.session.query(User).all()]
@view_config(route_name="view_user", renderer='json',
permission="view", request_method="GET")
def view_user_view(self):
user_id = self.request.matchdict['user_id']
user = self.session.query(User).filter(User.id==user_id).first()
if not user:
raise HTTPNotFound
return user.__json__()
@view_config(route_name="games", renderer='json',
permission="view", request_method="GET")
def list_games_view(self):
return [g.__json__() for g in self.session.query(Game).all()]
@view_config(route_name="games", renderer='json',
permission="edit", request_method="POST")
def new_game_view(self):
white = self.request.POST.get("white")
black = self.request.POST.get("black")
if not (white and
black and
self.session.query(User)
.filter(or_(User.id==white, User.id==black))
.count() == 2):
raise HTTPBadRequest
game = Game(board=chess.Game().fen(),
white_id=white, black_id=black)
self.session.add(game)
self.session.commit()
return game.__json__()
@view_config(route_name="view_game", renderer='json',
permission="view")
def view_game_view(self):
game_id = self.request.matchdict['game_id']
game = self.session.query(Game).filter(Game.id==game_id).first()
if not game:
raise HTTPNotFound
return game.__json__()
@view_config(route_name="make_move", renderer='json',
permission="edit", request_method="POST")
def make_move_view(self):
src = self.request.POST.get("src")
dst = self.request.POST.get("dst")
game_id = self.request.matchdict['game_id']
game = self.session.query(Game).filter(Game.id==game_id).first()
if not game:
raise HTTPNotFound
try:
game.move(src, dst)
except:
raise HTTPBadRequest
self.session.commit()
return game.__json__()
def __init__(self, request):
self.request = request
self.session = DBSession()
self.logged_in = authenticated_userid(request)