def init(self, user):
user_count = PubKeysLookup.load()
print('Loaded Trusted Directory of %d Public Keys.' % user_count)
self.prompt = ccolor.wrap(user) + '> '
pri_key = UserKeyLoader.pri_key(user)
pub_key = UserKeyLoader.pub_key(user)
print('Pri & Pub keys loaded for user ', ccolor.wrap(user))
self.u = User(user, pri_key, pub_key)
self.user_name = user
def register():
form = RegisterForm()
if form.validate_on_submit() and form.password_again.data == form.password.data:
db_session.global_init('db/mars_one.db')
session = db_session.create_session()
user = User()
user.email = form.email.data
user.password = form.password.data
user.city = form.city.data
session.add(user)
session.commit()
return redirect("/")
return render_template('register.html', form=form, current_user=None)
def get(self, id):
favorites = []
u = User.get_by_id(id)
if u is None:
abort(404, message="User not found")
for f in generate_sorted_query(Favorites).filter('user', u):
favorites.append(f)
return {'favorites': map(lambda f: marshal(f, favorite_user_fields), favorites)}
def get(self, id):
review_list = []
u = User.get_by_id(id)
if not u:
abort(404, message="User not found")
for r in generate_sorted_query(BeerReview).filter('user', u):
review_list.append(r)
return {'beer_reviews': map(lambda r: marshal(r, beer_review_fields), review_list)}
class CmdClient(cmd.Cmd):
def init(self, user):
user_count = PubKeysLookup.load()
print('Loaded Trusted Directory of %d Public Keys.' % user_count)
self.prompt = ccolor.wrap(user) + '> '
pri_key = UserKeyLoader.pri_key(user)
pub_key = UserKeyLoader.pub_key(user)
print('Pri & Pub keys loaded for user ', ccolor.wrap(user))
self.u = User(user, pri_key, pub_key)
self.user_name = user
def do_list(self, line):
print('List files of user : '******' [not shared] ' + ccolor.wrap(f_name) + '\t' +
str(int(files[f_name]['size'] / 1024)) + " KB")
# TODO: for each referenced group, retreived files
else:
print(' (empty) ')
end = timer()
print('Command execution time : %.2f sec' % (end - start))
def do_clear(self, line):
print('Deleting all content of user : '******'Command execution time : %.2f sec' % (end - start))
def do_upload(self, line):
files = line.split()
print('Uploading %s files for user : %s ' %
(ccolor.wrap(str(len(files))), ccolor.wrap(self.user_name)))
start = timer()
self.u.upload_files(files)
end = timer()
print('Command execution time : %.2f sec' % (end - start))
def do_download(self, line):
files = line.split()
print('Downloading %s files for user : %s ' %
(ccolor.wrap(str(len(files))), ccolor.wrap(self._user_id)))
start = timer()
self.u.download_files(files)
end = timer()
print('Command execution time : %.2f sec' % (end - start))
######################################################################
def do_group(self, line):
args = line.split()
if (args[0] == 'create'):
self.create_group(args[1:])
elif (args[0] == 'add'):
self.add_group_member()
else:
print('Unknown group command')
def create_group(self, args):
group_name = args[0]
group_members = args[1:]
print(
'Creating a new group ' + ccolor.group(group_name) +
' with members : ', group_members)
start = timer()
g = AdminGroupManagement(self.user_name)
g.create_group(group_name, group_members)
g.push_changes()
end = timer()
print('Command execution time : %.2f sec' % (end - start))
def add_group_member(self, args):
group_name = args[0]
group_members = args[1:]
print('Add to group ' + ccolor.group(group_name) + ' members : ',
group_members)
start = timer()
g = AdminGroupManagement(self.user_name)
g.add_members(group_name, group_members)
g.push_changes()
end = timer()
print('Command execution time : %.2f sec' % (end - start))
def do_exit(self, line):
return True
'/api/v1.0/beers/<int:id>/reviews',
endpoint='beer_reviews')
api.add_resource(BeerReviewListApi,
'/api/v1.0/beer_reviews',
endpoint='reviews')
api.add_resource(BeerReviewApi,
'/api/v1.0/beer_reviews/<int:id>',
endpoint='review')
api.add_resource(FavoritesUserApi,
'/api/v1.0/users/<int:id>/favorites',
endpoint='user_favorites')
api.add_resource(FavoritesBeerApi,
'/api/v1.0/beers/<int:id>/favorites',
endpoint='beer_favorites')
api.add_resource(FavoritesListApi, '/api/v1.0/favorites', endpoint='favorites')
api.add_resource(FavoritesApi,
'/api/v1.0/favorites/<int:id>',
endpoint='favorite')
# Create initial user if the user does not exist
u = User.all().get()
if u is None:
u = User(user_name='admin',
first_name='Admin',
last_name='Admin',
password=hash_password('beer_app1'))
u.put()
if __name__ == '__main__':
app.run(debug=True)
app = Flask(__name__)
api = Api(app)
api.add_resource(UserListApi, '/api/v1.0/users', endpoint='users')
api.add_resource(UserApi, '/api/v1.0/users/<int:id>', endpoint='user')
api.add_resource(BeerReviewUserApi, '/api/v1.0/users/<int:id>/reviews', endpoint='user_reviews')
api.add_resource(BeerGlassListApi, '/api/v1.0/beer_glasses', endpoint='beer_glasses')
api.add_resource(BeerGlassApi, '/api/v1.0/beer_glasses/<int:id>', endpoint='beer_glass')
api.add_resource(BeerListApi, '/api/v1.0/beers', endpoint='beers')
api.add_resource(BeerApi, '/api/v1.0/beers/<int:id>', endpoint='beer')
api.add_resource(BeerReviewBeerApi, '/api/v1.0/beers/<int:id>/reviews', endpoint='beer_reviews')
api.add_resource(BeerReviewListApi, '/api/v1.0/beer_reviews', endpoint='reviews')
api.add_resource(BeerReviewApi, '/api/v1.0/beer_reviews/<int:id>', endpoint='review')
api.add_resource(FavoritesUserApi, '/api/v1.0/users/<int:id>/favorites', endpoint='user_favorites')
api.add_resource(FavoritesBeerApi, '/api/v1.0/beers/<int:id>/favorites', endpoint='beer_favorites')
api.add_resource(FavoritesListApi, '/api/v1.0/favorites', endpoint='favorites')
api.add_resource(FavoritesApi, '/api/v1.0/favorites/<int:id>', endpoint='favorite')
# Create initial user if the user does not exist
u = User.all().get()
if u is None:
u = User(user_name='admin',
first_name='Admin',
last_name='Admin',
password=hash_password('beer_app1'))
u.put()
if __name__ == '__main__':
app.run(debug = True)
