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
