def create_company(cc_code, link="", name=""): company = Company() company.id = cc_code company.name = name company.link = link db.session.add(company) db.session.commit()
def register(): form = RegistrationForm() if form.validate_on_submit(): email = form.email.data username = form.username.data user = User(email=email, username=username) user.password = form.password.data company = Company() company.name = form.company.data db.session.add(user) db.session.add(company) company.set_company_owner(user) company.add_user(user) db.session.commit() company.company_materials_setup() db.session.commit() token = user.generate_confirmation_token() send_email( user.email, "Confirm Your Account", "auth/email/confirm", user=user, token=token, ) flash("A confirmation email has been sent to you by email.") return redirect(url_for("auth.login")) return render_template("auth/register.html", form=form)
def test_create_company(): c = Company() c.name = "test name" db.session.add(c) db.session.commit() db_c = Company.query.filter_by(name="test name").first() assert db_c == c
def test_users_company_name(clean_db): company = Company() company.name = "test name" user1 = User() user1.username = "******" db.session.add(user1) db.session.add(company) company.add_user(user1) db.session.commit() assert user1.company.name == company.name
def save(self, user, commit=True): try: company = Company() company.threshold = self.cleaned_data['threshold'] company.name = user.username try: list = json.loads(self.data["questions"]) company.num_of_questions = len(list) except: company.num_of_questions = 10 company.user = User.objects.get(pk=user.pk) if commit: company.save() return company except: raise AttributeError("invalid Parameters")
def stock_add(request): from app.remote.stocks import StockHistoryClient from app.forms import StockWizardForm from app.models import Company, Stock if request.method == "POST": form = StockWizardForm(request.POST) pprint(form.is_valid()) if form.is_valid(): # save company = Company() company.name = form.cleaned_data['company_name'] company.country = form.cleaned_data['company_country'] company.save() stock = Stock() stock.company = company stock.name = form.cleaned_data['name'] stock.wkn = form.cleaned_data['wkn'] stock.isin = form.cleaned_data['isin'] stock.symbol = form.cleaned_data['symbol'] stock.type = form.cleaned_data['type'] stock.default_stock_exchange = form.cleaned_data['default_stock_exchange'] stock.save() return HttpResponseRedirect(reverse('stock', args=(stock.symbol,))) else: pprint(form.errors) else: data = None if 'wkn_or_isin' in request.GET: shc = StockHistoryClient() data = shc.get_basics_by_wkn_or_isin(wkn_or_isin=request.GET['wkn_or_isin']) data['company_country'] = data['country'] data['company_name'] = data['name'] form = StockWizardForm(initial=data) return render_to_response( 'stock_wizard_add.html', { 'form': form }, context_instance=RequestContext(request) )
def test_company_owner(clean_db): name = "Test Company" company = Company() company.name = name user1 = User() user1.username = "******" db.session.add(user1) db.session.add(company) company.set_company_owner(user1) db.session.commit() db_company = Company.query.filter_by(name=name).first() assert user1 == db_company.owner
def _companies_from_json(app, content=''): if 'companies' in content: for obj in content['companies']: user = User.query.filter_by(email=obj['user']).first() if user is None: app.logger.error('Company: %s, Unknown user: %s' % (obj['name'], obj['user'])) break company = Company.query.filter_by(user=user, name=obj['name']).first() if company is None: company = Company() company.user = user company.name = obj['name'] company.description = obj['description'] app.logger.debug('adding company: %s' % company) db.session.add(company) db.session.commit()
def stock_add(request): from app.remote.stocks import StockHistoryClient from app.forms import StockWizardForm from app.models import Company, Stock if request.method == "POST": form = StockWizardForm(request.POST) pprint(form.is_valid()) if form.is_valid(): # save company = Company() company.name = form.cleaned_data['company_name'] company.country = form.cleaned_data['company_country'] company.save() stock = Stock() stock.company = company stock.name = form.cleaned_data['name'] stock.wkn = form.cleaned_data['wkn'] stock.isin = form.cleaned_data['isin'] stock.symbol = form.cleaned_data['symbol'] stock.type = form.cleaned_data['type'] stock.default_stock_exchange = form.cleaned_data[ 'default_stock_exchange'] stock.save() return HttpResponseRedirect(reverse('stock', args=(stock.symbol, ))) else: pprint(form.errors) else: data = None if 'wkn_or_isin' in request.GET: shc = StockHistoryClient() data = shc.get_basics_by_wkn_or_isin( wkn_or_isin=request.GET['wkn_or_isin']) data['company_country'] = data['country'] data['company_name'] = data['name'] form = StockWizardForm(initial=data) return render_to_response('stock_wizard_add.html', {'form': form}, context_instance=RequestContext(request))
def test_company_add_users(clean_db): company = Company() company.name = "test name" user1 = User() user1.username = "******" user2 = User() user2.username = "******" db.session.add(user1) db.session.add(user2) db.session.add(company) company.add_user(user1) company.add_user(user2) users = company.users assert user1 in users assert user2 in users
def test_saving_and_retrieving_models(self): company = Company() company.name = 'PSL' company.email = '*****@*****.**' company.telephone = 7777777 company.save() first_room = Room() first_room.name = 'Peter Santamaria' first_room.company = company first_room.security_level = 2 first_room.description = 'Conference room' first_room.id = 'r00m1' first_room.save() second_room = Room() second_room.name = '201' second_room.company = company second_room.security_level = 1 second_room.description = 'Class room' second_room.id = 'r00m2' second_room.save() third_room = Room() third_room.name = 'rest' third_room.company = company third_room.security_level = 1 third_room.description = 'Rest room' third_room.id = 'r00m3' third_room.save() event = Event() event.name = 'My event' event.company = company event.start_date = datetime(2017, 3, 16) event.end_date = datetime(2017, 3, 21) event.event_id = '3v3nt' event.rooms.add(first_room) event.rooms.add(second_room) event.rooms.add(third_room) event.save() enduser = EndUser() enduser.id = 'u53r' enduser.name = 'Sebastian' enduser.last_name = 'Villegas' enduser.email = '*****@*****.**' enduser.save() first_permission = Permission() first_permission.user_id = enduser first_permission.event = event first_permission.id = '93rm151' first_permission.save() second_permission = Permission() second_permission.user_id = enduser second_permission.event = event second_permission.id = '93rm152' second_permission.save() saved_company = Company.objects.first() self.assertEqual(saved_company, company) saved_rooms = Room.objects.all() self.assertEqual(saved_rooms.count(), 3) self.assertEqual(saved_rooms[0], first_room) self.assertEqual(saved_rooms[1], second_room) self.assertEqual(saved_rooms[2], third_room) saved_event = Event.objects.first() self.assertEqual(saved_event, event) saved_enduser = EndUser.objects.first() self.assertEqual(saved_enduser, enduser) saved_permissions = Permission.objects.all() self.assertEqual(saved_permissions.count(), 2) self.assertEqual(saved_permissions[0], first_permission) self.assertEqual(saved_permissions[1], second_permission)
def get_company_node(company_name): if company_name is not None: company = Company() company.name = company_name return company