def save_employee_data(request, employee_data, line_count):
line_count = line_count + 1
if User.objects.filter(username__iexact=employee_data['email']).exists():
messages.error(
request,
f"Error! Line: {line_count} {employee_data['email']} email is taken"
)
return redirect('upload_employees')
elif User.objects.filter(email__iexact=employee_data['email']).exists():
messages.error(
request,
f"Error! Line: {line_count} {employee_data['email']} email is taken"
)
return redirect('upload_employees')
else:
# Store user detals
email = employee_data['email'].lower()
user = User.objects.create_user(first_name=employee_data['first_name'],
last_name=employee_data['last_name'],
username=employee_data['email'],
password=employee_data['password'],
email=employee_data['email'])
user.save()
companyObj = None
if Company.objects.filter(name=employee_data['company']).exists():
companyObj = Company.objects.get(name=employee_data['company'])
else:
companyObj = Company(name=employee_data['company'])
companyObj.save()
profile = Profile(user=user, company=companyObj)
profile.save()
return
def register(request):
if request.method == "POST":
try:
user = User.objects.create_user(
email=request.POST['email'],
password=request.POST['password'],
)
user.first_name = request.POST['first_name']
user.last_name = request.POST['last_name']
user.save()
profile = Profile()
profile.user = user
profile.save()
is_employer = (request.POST['is_employer'] == '0')
if (is_employer):
company = Company()
company.user = user
company.save()
return render(request, 'registration/edit.html',
{'is_company': is_employer})
except:
return render(request, 'registration/edit.html',
{'is_company': False})
else:
return render(request, 'registration/registration.html')
def create_user(self, username, password, company_name):
user = User.objects.create(username=username)
user.set_password(password)
user.save()
company = Company(Name=company_name)
company.Owned_by = user
company.save()
company_user = CompanyUser(User_id=user.id, Company_id=company.id)
company_user.save()
self.stdout.write(f'User {username} created with password {password}')
def create(request):
print(request.POST)
company_name = request.POST.get("name")
company_phone = request.POST.get("phone")
company_email = request.POST.get("email")
company_object = Company(name=company_name,
phone=company_phone,
email=company_email)
company_object.save()
return redirect("companies:index")
def set(request):
print("hit endpoint")
data = app.get_companies()
existing = Company.objects.all()
names = []
for existing_company in existing:
names.append(existing_company.name)
for item in data:
if item[1] not in names:
company = Company(name=item[1], ticker=item[0], industry=item[2])
company.save()
return HttpResponse("Success...?")
def create_company(cls, company_data):
updates = {}
company_key = firebase_db.reference('companies').push().key
company = Company(
external_key=company_key,
**company_data,
)
company.save()
updates['companies/{key}/pk'.format(key=company_key)] = company.pk
firebase_db.reference().update(updates)
return company
def setUp(self):
company = Company(company_name='Name')
worker = Worker(name='Worker')
work = Work(
company=company,
work_name='WorkName',
)
self.wp = WorkPlace(
worker=worker,
work=work,
workplace_name='new workplace',
)
def post(self, request, format=None):
company_instance = Company(
name=request.data['name'],
short_name=request.data['short_name'],
inn=request.data['inn'],
active=True,
)
badge = request.FILES.get('badge', None)
if not badge is None:
company_instance.badge = badge
company_instance.save()
company_instance.user.add(request.user)
return Response(
CompanySerializer(company_instance).data, status.HTTP_201_CREATED)
def create(self, request, *args, **kwargs):
# get company
company = request.data.pop('company')
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
user = serializer.instance
# save company
c = Company(company_name=company)
c.generate_code()
c.save()
c.staff.add(user)
return Response(serializer.data, status=status.HTTP_201_CREATED)
def create(request, data):
if request.method == 'POST':
# Collect form data.
company_name = request.POST['company_name']
owned_by = Staff.objects.get(user=request.user)
# Create new Company instance.
new_company = Company(name=company_name, owner=owned_by)
new_company.save()
# Add new company to workplaces.
owned_by.workplaces.add(new_company)
# Add owner to staff of new company.
new_company.staff.add(owned_by)
# Success
data['alerts'].append(('success', 'Create successfully!',
'You have successfully create a new company.'))
return redirect_with_data(request, data, '/companies/1+1/#tab2')
else:
return render(request, 'companies/create.html', data)
def _fill_companies() -> None:
company_logos_directory = Path('company_logos')
company_names: List[str] = [
'Workiro',
'Rebel Rage',
'Staffing Smarter',
'Evilthreat',
'Hirey',
'SwiftAttack',
'TROLLER',
'Primal Assault',
]
for index, companies_name in enumerate(company_names):
owner: LaborExchangeUser
is_created: bool
owner, is_created = LaborExchangeUser.objects.get_or_create(
name=f'Владелец',
surname=f'Хозяинов_{index}',
email=f'owner{index}@owner.ru',
phone=f'8999777665{index}',
)
assert is_created, 'Admin was not created'
owner.set_password(f'Djangotest{index}')
owner.save()
company_instance = Company(
name=companies_name,
location='Самый лучший город на Земле',
logo=str(company_logos_directory / f'b003a3c{index + 1}.png'),
owner=owner,
employee_count=choice(range(1, 300)),
description=
'Это компания без описания, потому что её HR-специалисты — ленивые задницы',
)
company_instance.save()
def handle(self, *args, **options):
data = csv.reader(open(
os.getcwd() + '/companies/management/commands/CCFInfo3.csv', 'rU'),
quotechar='"',
delimiter=',')
# print (data)
for row in data:
# print "0"+row[0]
new_company = Company(name=row[0],
callisto_url=row[1],
website=row[2],
logo=row[3],
expected_hires=row[4],
number_employees=row[5],
number_domestic_locations=row[6],
number_international_locations=row[7],
organization_type=row[8],
company_description=row[9],
job_description=row[10],
position_locations=row[11],
position_types=row[12])
new_company.save()
self.stdout.write('Success')
from companies.models import Company, Domain
companies_info = [
{
'schema_name': 'ford',
'name': 'Ford',
},
{
'schema_name': 'mazda',
'name': 'Mazda',
},
{
'schema_name': 'chevrolet',
'name': 'Chevrolet',
},
{
'schema_name': 'bmw',
'name': 'BMW',
}
]
for company in companies_info:
company_object = Company(schema_name=company['schema_name'], name=company['name'])
company_object.save()
domain = Domain()
domain.domain = '%s.localhost' % company['schema_name']
domain.tenant = company_object
domain.is_primary = True
domain.save()
"Systems": "orange",
"Data": "yellow",
}
for t in colors.keys():
industry = Industry(name=t, color=colors[t])
industry.save()
industries.append(industry)
print("Generating companies...")
companies = []
for i in range(10):
company = Company(
name=fake.company(),
location=fake.address(),
status=random.choice(Company.STATUSES)[0],
size=random.choice(Company.SIZES)[0],
donated=random.randint(0, 10000),
updated=fake.date_time_this_month(before_now=True,
after_now=False,
tzinfo=None),
)
company.save()
for s in random.sample(industries, random.randint(1, 3)):
company.industries.add(s)
company.save()
companies.append(company)
print("Generating contacts...")
contacts = []
for i in range(20):
contact = Contact(
first_name=fake.first_name(),
"Software": "purple",
"Systems": "orange",
"Data": "yellow",
}
for t in colors.keys():
industry = Industry(name=t, color=colors[t])
industry.save()
industries.append(industry)
print("Generating companies...")
companies = []
for i in range(50):
company = Company(
name=fake.company(),
location=fake.address(),
size=random.choice(Company.SIZES)[0],
updated=fake.date_time_this_month(before_now=True,
after_now=False,
tzinfo=None),
)
company.save()
for s in random.sample(industries, random.randint(1, 3)):
company.industries.add(s)
company.save()
companies.append(company)
print("Generating contacts...")
contacts = []
for i in range(150):
contact = Contact(
first_name=fake.first_name(),
last_name=fake.last_name(),
# Created by Adam Jacobs, June 2017 from django.contrib.auth.models import User from exhibitors import models from orders.models import Order, Product, ProductType from fair.models import Fair from companies.models import Company, Contact from datetime import date # Get super user u = User.objects.get() fair = Fair(name="fair1", year=2017, description="description", current=True ) fair.save() company = Company(name="TestCompany", organisation_type='company') company.save() contact = Contact(user=u, belongs_to=company, name="contact name", email="*****@*****.**", active=True, confirmed=True) contact.save() banquetType = ProductType(name="Banquet", description="...") banquetType.save() lunchType = ProductType(name="AdditionalLunch", description="...") lunchType.save() eventsType = ProductType(name="Events", description="...") eventsType.save() roomsType = ProductType(name="Rooms", description="...") roomsType.save() novaType = ProductType(name="Nova", description="...") novaType.save()
def main_parse(search,
page,
quantity,
vacancy_on_page=10,
location=None,
country=None):
country_site = IndeedCountrySites.objects.get(code_iso=country)
start_at = page * vacancy_on_page
debug_log(f'{"#"* 21}\nParse starts at: {start_at}')
total_company_add = 0
total_vacancies_add = 0
for start in range(start_at, quantity, vacancy_on_page):
page_num = int(start / vacancy_on_page)
debug_log(f"Parse Page: {page_num}")
params = {'start': start, 'sort': settings.DEFAULT_SORT}
if search:
params.update({'q': search})
if location:
params.update({'l': location})
debug_log(f'Params: {params}')
try:
vacancies = get_vacancies_on_page(country=country,
session=session,
params=params)
if vacancies is None:
continue
company_add = 0
vacancies_add = 0
for vacancy_data in vacancies:
company_uid = vacancy_data.get('cmpid', None)
company_link = vacancy_data.get('company_link', None)
if company_uid:
company = Company.objects.filter(uid=company_uid).first()
if not company and company_link:
_counry = IndeedCountrySites.objects.get(
code_iso=country)
company_link = normalize_url(_counry.site,
company_link)
company_data = get_company_info(session, company_link)
if company_data:
company = Company(
uid=company_uid,
name=company_data.get('title'),
link=company_link,
about=company_data.get('about'),
headquarters=company_data.get('headquarters'),
employees=company_data.get('employees'),
industry=company_data.get('industry'),
website=company_data.get('website'),
revenue=company_data.get('revenue'),
country_site=country_site)
company.save()
company_add += 1
else:
company = None
else:
company = None
job_key = vacancy_data.get('jk')
if not job_key:
continue
exist_vacancy = Vacancy.objects.filter(job_key=job_key).first()
print(exist_vacancy)
if exist_vacancy:
continue
vacancy_url = vacancy_data.get('link')
vacancy = Vacancy(
job_key=job_key,
title=vacancy_data.get('title'),
link=normalize_url(country_site.site, vacancy_url),
description_text=vacancy_data.get('description',
{}).get('text'),
description_html=vacancy_data.get('description',
{}).get('html'),
location=vacancy_data.get('loc'),
location_uid=vacancy_data.get('locid'),
country=vacancy_data.get('country'),
city=vacancy_data.get('city'),
zip=vacancy_data.get('zip'),
company_name=url_parse.unquote(
vacancy_data.get('srcname', '')),
company_uid=vacancy_data.get('cmpid'),
company_link=normalize_url(country_site.site, company_link)
if company_link else None,
company_id=company.id if company else None,
raw_data=vacancy_data,
country_site=country_site)
_data = {
'title': vacancy.title,
'link': vacancy.link,
'description_text': vacancy.description_text,
'description_html': vacancy.description_html,
'location': vacancy.location,
'location_uid': vacancy.location_uid,
'country': vacancy.company,
'city': vacancy.city,
'zip': vacancy.zip,
'company_name': vacancy.company_name,
'company_uid': vacancy.company_uid,
'company_link': vacancy.company_link,
'company_id': company.id if company else None,
'raw_data': vacancy.raw_data,
'country_site': country_site
}
Vacancy.objects.get_or_create(job_key=job_key, defaults=_data)
vacancies_add += 1
total_company_add += company_add
total_vacancies_add += vacancies_add
debug_log(
f"TC: {total_company_add}, TV: {total_vacancies_add}, C: {company_add}, V: {vacancies_add}"
)
except Exception as e:
tb = traceback.format_exc()
error_log(f"###\nPage {page}\n{tb}")
raise
def save(self):
self.clean()
cleaned_data = self.cleaned_data
userManager = MyUserManager()
email = cleaned_data.get('email')
firstName = cleaned_data.get('firstName')
lastName = cleaned_data.get('lastName')
user_type = None
if self.is_employer_selected():
user_type = USER_TYPE_EMPLOYER
else:
user_type = USER_TYPE_CANDIDATE
password = cleaned_data.get('password')
print(user_type)
user = User()
user.email = email
user.firstName = firstName
user.lastName = lastName
user.user_type = user_type
user.set_password(password)
user.save()
if cleaned_data.get('preferredName') != '' and cleaned_data.get(
'preferredName') != None:
preferredName = PreferredName()
preferredName.user = user
preferredName.preferredName = cleaned_data.get('preferredName')
preferredName.save()
if self.is_employer_selected():
employer = Employer()
employer.user = user
if self.is_createCompany_selected():
company = Company()
company.name = cleaned_data.get('companyName')
company.address = cleaned_data.get('address')
company.website = cleaned_data.get('website')
company.profile = cleaned_data.get('profile')
company.image = cleaned_data.get('image')
company.save()
employer.company = company
else:
employer.company = get_object_or_404(
Company, pk=cleaned_data.get('company'))
employer.save()
else:
candidate = Candidate()
candidate.user = user
candidate.studentID = cleaned_data.get('studentID')
candidate.creditCompleted = cleaned_data.get('creditCompleted')
candidate.program = cleaned_data.get('program')
candidate.creditLeft = cleaned_data.get('creditLeft')
candidate.gpa = cleaned_data.get('gpa')
candidate.internationalStudent = cleaned_data.get(
'internationalStudent')
candidate.travel = cleaned_data.get('travel')
candidate.timeCommitment = cleaned_data.get('timeCommitment')
candidate.transcript = cleaned_data.get('transcript')
candidate.save()
for lan in self.languageFieldsNames:
language = Language()
language.language = cleaned_data.get(lan['language'])
language.fluency = cleaned_data.get(lan['proficiency'])
language.details = cleaned_data.get(lan['details'])
language.user = user
language.save()
return user
def register(request):
if request.method == 'POST':
# Retrieving all details from POST request
first_name = request.POST['first_name']
last_name = request.POST['last_name']
# Setting the email as the username
username = request.POST['email']
email = request.POST['email']
password = request.POST['password']
password2 = request.POST['password2']
company_count = Company.objects.all().count()
# Adding new company if no companies currently stored
if company_count == 0:
if request.POST['newCompanyInput'] == '':
messages.error(request, 'Please enter a company name')
return redirect('register')
company_name = request.POST['newCompanyInput']
else:
# Checking if a new company was entered and storing it
if 'newCompanyInput' in request.POST and request.POST[
'newCompanyInput'] != '':
company_name = request.POST['newCompanyInput']
else:
company_name = request.POST['company']
if password == password2:
# Check username
if ' ' in username:
messages.error(request, 'Email cannot have spaces')
return redirect('register')
if User.objects.filter(username__iexact=username).exists():
messages.error(request, 'That email is taken')
return redirect('register')
elif User.objects.filter(email__iexact=email).exists():
messages.error(request, 'That email is being used')
return redirect('register')
else:
# Store user detals
email = email.lower()
user = User.objects.create_user(first_name=first_name,
last_name=last_name,
username=username,
password=password,
email=email)
user.save()
if Company.objects.filter(name=company_name).exists():
companyObj = Company.objects.get(name=company_name)
else:
companyObj = Company(name=company_name)
companyObj.save()
profile = Profile(user=user, company=companyObj)
profile.save()
messages.success(request,
'You are now registered and can login')
return redirect('login')
else:
messages.error(request, 'Passwords do not match')
return redirect('register')
else:
# Retrieving the list of companies to populate the dropdown during registration
companies = Company.objects.all()
context = {"companies": companies}
return render(request, 'accounts/register.html', context)
def test_company(self):
company = Company(company_name='Name')
self.assertIsNotNone(company)
def test_list_all_companies(db_session):
repository = CompanyRepository(db_session)
repository.save(Company(id=generate_id(), name="Pepsi"))
result = repository.list_all()
assert len(result) == 1
def create_company(name):
C = Company(name=name)
C.save()
return C
from companies.models import Company, Domain
# create your public tenant
tenant = Company(schema_name='public',
name='Colombia Cars')
tenant.save()
# Add one or more domains for the tenant
domain = Domain()
domain.domain = 'localhost'
domain.tenant = tenant
domain.is_primary = True
domain.save()
