def post(self): name = self.request.get('name') if Company.query(Company.name == name).count() > 0: response = {'success': False, 'error_message': "A company already exists with the name {0}".format(name)} self.response.write(json.dumps(response)) return new_company = Company.create_company(name) _ = CompanyAccount.create_company_account(company=new_company, user_account=self.user_account) response = {'success': True, 'goto_url': '/account/company/{0}'.format(new_company.key.id())} self.response.write(json.dumps(response))
def get(self): company_account_query = CompanyAccount.query(CompanyAccount.user_account_key == self.user_account.key) companies = [company_account.company for company_account in company_account_query] self.template_values['user_companies'] = companies self.render_template('account/index.html')