def _create_user(request, mysqldb):
from ogreserver.models.user import User
# create random username
username = "".join(random.choice(string.ascii_lowercase) for n in range(6))
# create user in auth DB
user = User(username, password=username, email="{}@example.com".format(username), active=True, roles=[])
user.preferred_ebook_format = "mobi"
mysqldb.add(user)
mysqldb.commit()
return user
def post():
try:
# authenticate user
user = User.validate_auth_key(username=request.form.get("username"),
api_key=request.form.get("api_key"))
if user is None:
return "API Auth Failed: Post"
# decode the json payload
data = json.loads(request.form.get("ebooks"))
except KeyError:
return "Bad Request"
# stats log the upload
Log.create(user.id, "CONNECT", request.form.get("total"),
request.form.get("api_key"))
# update the library
ds = DataStore(user)
new_ebook_count = ds.update_library(data)
Log.create(user.id, "NEW", new_ebook_count, request.form.get("api_key"))
# handle badge and reputation changes
r = Reputation(user)
r.new_ebooks(new_ebook_count)
r.earn_badges()
msgs = r.get_new_badges()
# query books missing from S3 and supply back to the client
rs = ds.find_missing_books()
return json.dumps({'ebooks_to_upload': rs, 'messages': msgs})
def auth():
user = User.authenticate(username=request.form.get("username"),
password=request.form.get("password"))
if not user:
return "0"
else:
return user.assign_auth_key()
def validate(self):
if not super(LoginForm, self).validate():
return False
user = User.authenticate(username=self.username.data, password=self.password.data)
if not user:
self.username.errors.append('Invalid details.')
return False
self.user = user
return True
def validate(self):
if not super(LoginForm, self).validate():
return False
user = User.authenticate(username=self.username.data,
password=self.password.data)
if not user:
self.username.errors.append('Invalid details.')
return False
self.user = user
return True
def upload():
user = User.validate_auth_key(
username=request.form.get("username"),
api_key=request.form.get("api_key")
)
if user is None:
return "API Auth Failed: Upload"
# stats log the upload
Log.create(user.id, "UPLOAD", 1, request.form.get("api_key"))
uploads.save(request.files['ebook'], None, "%s%s" % (request.form.get("filehash"), request.form.get("format")))
res = store_ebook.apply_async((user.id, request.form.get("sdbkey"), request.form.get("filehash"), request.form.get("format")))
return res.task_id
def upload():
user = User.validate_auth_key(username=request.form.get("username"),
api_key=request.form.get("api_key"))
if user is None:
return "API Auth Failed: Upload"
# stats log the upload
Log.create(user.id, "UPLOAD", 1, request.form.get("api_key"))
uploads.save(
request.files['ebook'], None,
"%s%s" % (request.form.get("filehash"), request.form.get("format")))
res = store_ebook.apply_async(
(user.id, request.form.get("sdbkey"), request.form.get("filehash"),
request.form.get("format")))
return res.task_id
def post():
try:
# authenticate user
user = User.validate_auth_key(
username=request.form.get("username"),
api_key=request.form.get("api_key")
)
if user is None:
return "API Auth Failed: Post"
# decode the json payload
data = json.loads(request.form.get("ebooks"))
except KeyError:
return "Bad Request"
# stats log the upload
Log.create(user.id, "CONNECT", request.form.get("total"), request.form.get("api_key"))
# update the library
ds = DataStore(user)
new_ebook_count = ds.update_library(data)
Log.create(user.id, "NEW", new_ebook_count, request.form.get("api_key"))
# handle badge and reputation changes
r = Reputation(user)
r.new_ebooks(new_ebook_count)
r.earn_badges()
msgs = r.get_new_badges()
# query books missing from S3 and supply back to the client
rs = ds.find_missing_books()
return json.dumps({
'ebooks_to_upload': rs,
'messages': msgs
})
def auth():
user = User.authenticate(username=request.form.get("username"), password=request.form.get("password"))
if not user:
return "0"
else:
return user.assign_auth_key()
