def signUp(username, userid): CONST_ERROR_ALR_REG = 0 CONST_OPTION_REG_SUCCESS = 1 user_query = (User.query()).fetch() # Get all users in the datastore # If this user already exists if (any(q.user_id == userid for q in user_query)): return CONST_ERROR_ALR_REG # Else if this user has not been registered else: new_user = User(user_name=str(username), user_id=userid) new_user.put() return CONST_OPTION_REG_SUCCESS
def post(self):
"""Function that gets called when form is submitted"""
error_username = ""
error_password = ""
error_email = ""
entered_username = cgi.escape(self.request.get("username"))
if cgi.escape(self.request.get("username")) == "" :
error_username = "******"
elif verify_username(entered_username) == None:
error_username = "******"
else :
query= db.GqlQuery("select * from User where username = :1",
entered_username)
if query.get():
error_username = "******"
entered_password = cgi.escape(self.request.get("password"))
entered_verify = cgi.escape(self.request.get("verify"))
if entered_password == "":
error_password = "******"
elif entered_verify == "":
error_password = "******"
elif entered_password != entered_verify :
error_password = "******"
entered_email = cgi.escape(self.request.get("email"))
if entered_email and verify_email(entered_email) == None:
error_email = "Email not valid"
if (error_username != "" or error_password !="" or error_email!=""):
self.show_form(error_username,
error_password,
error_email,
entered_username,
entered_email)
else:
new_user = User(username=entered_username,
hashed_pwd=make_hashed_password(entered_password),
email=entered_email)
new_user.put()
self.set_secure_cookie('user_id',str(new_user.key().id()))
self.redirect("/profile")
def post(self):
"""Check if the username is already taken. If not create a new user."""
users = db.GqlQuery("select * from User order by created desc")
username = self.request.get('username')
password = self.request.get('password')
verify = self.request.get('verify')
have_error = False
error_msg = ""
# check if all values are valid input, and passwords match
if not valid_username(username):
error_msg = "Username must be between 3 and 20 alphanumeric characters."
have_error = True
else:
for user in users:
if user.username == username:
error_msg = "Username is already in use."
have_error = True
if not valid_password(password):
error_msg = "Password must be between 3 and 20 characters."
have_error = True
if not password == verify:
error_msg = "Passwords do not match."
have_error = True
# if something is invalid, render page again with error message
if have_error:
self.render('register.html', username=username, error_msg=error_msg)
else:
#otherwise, create a new user, and redirect to main page
user = User(username=username, password=password)
user.put()
username = str(username)
self.response.headers.add_header('Set-Cookie', 'user_id=%s; Path=/' % username)
self.redirect('/')
def update_database(data):
print('updating db')
# start upload text
if int(data['rows_saved']) > 0:
if 'saved' in data:
saved = data['saved']
if 'addresses' in saved:
for addresses in saved['addresses']:
address = Address()
where = {'id': addresses['id']}
data = {'address_id': addresses['address_id']}
address.put(where=where, data=data)
if 'colors' in saved:
for colors in saved['colors']:
color = Colored()
where = {'id': colors['id']}
data = {'color_id': colors['color_id']}
color.put(where=where, data=data)
if 'cards' in saved:
for cards in saved['cards']:
card = Card()
where = {'id': cards['id']}
data = {'card_id': cards['card_id']}
card.put(where=where, data=data)
if 'credits' in saved:
for credit in saved['credits']:
credits = Credit()
where = {'id': credit['id']}
data = {'credit_id': credit['credit_id']}
credits.put(where=where, data=data)
if 'companies' in saved:
for companies in saved['companies']:
company = Company()
where = {'id': companies['id']}
data = {'company_id': companies['company_id']}
company.put(where=where, data=data)
if 'custids' in saved:
for custids in saved['custids']:
custid = Custid()
where = {'id': custids['id']}
data = {'cust_id': custids['cust_id']}
custid.put(where=where, data=data)
if 'deliveries' in saved:
for deliveries in saved['deliveries']:
delivery = Delivery()
where = {'id': deliveries['id']}
data = {'delivery_id': deliveries['delivery_id']}
delivery.put(where=where, data=data)
if 'discounts' in saved:
for discounts in saved['discounts']:
discount = Discount()
where = {'id': discounts['id']}
data = {'discount_id': discounts['discount_id']}
discount.put(where=where, data=data)
if 'inventories' in saved:
for inventories in saved['inventories']:
inventory = Inventory()
where = {'id': inventories['id']}
data = {'inventory_id': inventories['inventory_id']}
inventory.put(where=where, data=data)
if 'inventory_items' in saved:
for inventory_items in saved['inventory_items']:
inventory_item = InventoryItem()
where = {'id': inventory_items['id']}
data = {'item_id': inventory_items['item_id']}
inventory_item.put(where=where, data=data)
if 'invoices' in saved:
for invoices in saved['invoices']:
invoice = Invoice()
where = {'id': invoices['id']}
data = {'invoice_id': invoices['invoice_id']}
invoice.put(where=where, data=data)
if 'invoice_items' in saved:
for invoice_items in saved['invoice_items']:
invoice_item = InvoiceItem()
where = {'id': invoice_items['id']}
data = {'invoice_items_id': invoice_items['invoice_items_id']}
invoice_item.put(where=where, data=data)
if 'memos' in saved:
for memos in saved['memos']:
memo = Memo()
where = {'id': memos['id']}
data = {'memo_id': memos['memo_id']}
memo.put(where=where, data=data)
if 'printers' in saved:
for printers in saved['printers']:
printer = Printer()
where = {'id': printers['id']}
data = {'printer_id': printers['printer_id']}
printer.put(where=where, data=data)
if 'profiles' in saved:
for profiles in saved['profiles']:
profile = Profile()
where = {'id': profiles['id']}
data = {'p_id': profiles['p_id']}
profile.put(where=where, data=data)
if 'reward_transactions' in saved:
for reward_transactions in saved['reward_transactions']:
reward_transaction = RewardTransaction()
where = {'id': reward_transactions['id']}
data = {'reward_id': reward_transactions['reward_id']}
reward_transaction.put(where=where, data=data)
if 'rewards' in saved:
for rewards in saved['rewards']:
reward = Reward()
where = {'id': rewards['id']}
data = {'reward_id': rewards['reward_id']}
reward.put(where=where, data=data)
if 'schedules' in saved:
for schedules in saved['schedules']:
schedule = Schedule()
where = {'id': schedules['id']}
data = {'schedule_id': schedules['schedule_id']}
schedule.put(where=where, data=data)
if 'taxes' in saved:
for taxes in saved['taxes']:
tax = Tax()
where = {'id': taxes['id']}
data = {'tax_id': taxes['tax_id']}
tax.put(where=where, data=data)
if 'transactions' in saved:
for transactions in saved['transactions']:
transaction = Transaction()
where = {'id': transactions['id']}
data = {'trans_id': transactions['trans_id']}
transaction.put(where=where, data=data)
if 'users' in saved:
for users in saved['users']:
user = User()
where = {'id': users['id']}
data = {'user_id': users['user_id']}
user.put(where=where, data=data)
if 'zipcodes' in saved:
for zipcodes in saved['zipcodes']:
zipcode = Zipcode()
where = {'id': zipcodes['id']}
data = {'zipcode_id': zipcodes['zipcode_id']}
zipcode.put(where=where, data=data)
