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)