def post(self, request):
form = CompanyForm(request.POST, request.FILES)
if not form.is_valid():
return render(request, 'vacancies/user/company_edit.html', {'form': form})
company = self.get_company(request)
if company:
company.title = form.cleaned_data['title']
company.location = form.cleaned_data['location']
if form.cleaned_data['logo']:
company.logo = form.cleaned_data['logo']
company.description = form.cleaned_data['description']
company.employee_count = form.cleaned_data['employee_count']
company.owner = get_user(request)
else:
company = Company(
title=form.cleaned_data['title'],
location=form.cleaned_data['location'],
logo=form.cleaned_data['logo'],
description=form.cleaned_data['description'],
employee_count=form.cleaned_data['employee_count'],
owner=get_user(request),
)
company.save()
return redirect(reverse('my_company_form') + '?submitted=True')
def post(self, request):
user = request.user
if user.is_anonymous:
return redirect(reverse('login'))
company = Company.objects.filter(owner=user).first()
if company is None:
company = Company()
form = CompanyForm(request.POST)
context = {'form': form, 'company': company}
if form.is_valid():
data = form.cleaned_data
company.name = data['name']
company.location = data['location']
# Не знаю, как надо загружать logo
# в request.POST данные есть, а в cleaned_data - нет
# company.logo = data['logo']
#
# file_name = request.POST.get('logo')
# company.logo.storage.save(file_name)
company.description = data['description']
company.employee_count = data['employee_count']
company.owner = user
company.save()
context['info_updated'] = True
return render(self.request, 'vacancies/company-edit.html', context)
def main():
# companies
Company.objects.all().delete()
for company in data.companies:
new_company = Company(
id=company['id'],
name=company['title'],
location=company['location'],
logo=company['logo'],
description=company['description'],
employee_count=company['employee_count'],
)
new_company.save()
# specialties
Specialty.objects.all().delete()
for specialty in data.specialties:
new_specialty = Specialty(
code=specialty['code'],
title=specialty['title'],
picture='specty_' + specialty['code'] + '.png',
)
new_specialty.save()
# vacancies
Vacancy.objects.all().delete()
for job in data.jobs:
new_vacancy = Vacancy(
id=job['id'],
title=job['title'],
specialty=Specialty.objects.get(code=job['specialty']),
company=Company.objects.get(id=job['company']),
skills=job['skills'],
description=job['description'],
salary_min=job['salary_from'],
salary_max=job['salary_to'],
published_at=job['posted'],
)
new_vacancy.save()
# settings
Settings.objects.all().delete()
new_setting = Settings(name='site_title', setval='Джуманджи')
new_setting.save()
new_setting = Settings(name='site_description',
setval='Вакансии для <br>Junior-разработчиков')
new_setting.save()
def fill_companies(companies):
"""Populate company database
Args:
companies (dict): 'companies' dict from moc-file
Returns:
dict: mapping dictionary for saving job->company relations
"""
mapping = {}
for company in companies:
moc_id = company.pop("id")
company["name"] = company["title"]
_ = company.pop("title")
new_company = Company(**company)
new_company.save()
mapping.setdefault(moc_id, new_company.pk)
return mapping
def handle(self, *args, **options):
for specialty in specialties:
sp = Specialty(code=specialty['code'],
title=specialty['title'],
picture='https://place-hold.it/100x60')
sp.save()
for company in companies:
cm = Company(name=company['title'],
logo='https://place-hold.it/100x60')
cm.save()
for job in jobs:
vc = Vacancy(
title=job['title'],
specialty=Specialty.objects.filter(code=job['cat']).first(),
company=Company.objects.filter(name=job['company']).first(),
salary_min=job['salary_from'],
salary_max=job['salary_to'],
published_at=job['posted'],
description=job['desc'])
vc.save()
if __name__ == '__main__':
for spec in data.specialties:
specialty = Specialty(code=spec['code'],
title=spec['title'],
picture=f'{settings.MEDIA_SPECIALITY_IMAGE_DIR}/specty_{spec["code"]}.png')
specialty.save()
loc = ['Москва', 'Калуга', 'п. Марс', 'Брянск', 'Прага', 'Лос-Анджелес', 'Сухиничи']
for index, company in enumerate(data.companies, start=1):
company = Company(name=company['title'],
location=random.choice(loc),
logo=f'{settings.MEDIA_COMPANY_IMAGE_DIR}/logo{index}.png',
description='',
employee_count=random.randint(1, 1000))
company.save()
skills = ['Python', 'Django', 'Git', 'Linux', 'CSS', 'Excel', 'Photoshop', 'ICQ', 'C++', 'Java', 'C#']
for vacancy in data.jobs:
specialty = Specialty.objects.get(code=vacancy['cat'])
company = Company.objects.get(name=vacancy['company'])
_vacancy = Vacancy(title=vacancy['title'],
specialty=specialty,
company=company,
skills=', '.join(random.sample(skills, 4)),
description=vacancy['desc'],
salary_min=int(vacancy['salary_from']),
salary_max=int(vacancy['salary_to']),
published_at=vacancy['posted'])
_vacancy.save()
def main():
# users
if not User.objects.all().count():
last_login = '******'
User.objects.create_superuser(username='******',
email='admin@stepik_vacancies.com',
password='******',
last_login=last_login,
first_name='Иван',
last_name='Петров')
User.objects.create_user(username='******',
email='user1@stepik_vacancies.com',
password='******',
last_login=last_login,
first_name='Петр',
last_name='Сергеев')
User.objects.create_user(username='******',
email='user2@stepik_vacancies.com',
password='******',
last_login=last_login,
first_name='Сергей',
last_name='Иванов')
# companies
users = User.objects.all()
Company.objects.all().delete()
for company in data.companies:
user = users[randint(0, 2)]
new_company = Company(
id=company['id'],
name=company['title'],
location=company['location'],
description=company['description'],
employee_count=company['employee_count'],
owner=user,
)
new_company.save()
# specialties
Specialty.objects.all().delete()
for specialty in data.specialties:
new_specialty = Specialty(
code=specialty['code'],
title=specialty['title'],
)
new_specialty.save()
# vacancies
Vacancy.objects.all().delete()
for job in data.jobs:
new_vacancy = Vacancy(
id=job['id'],
title=job['title'],
specialty=Specialty.objects.get(code=job['specialty']),
company=Company.objects.get(id=job['company']),
skills=job['skills'],
description=job['description'],
salary_min=job['salary_from'],
salary_max=job['salary_to'],
published_at=job['posted'],
)
new_vacancy.save()