def save(self): data=self.cleaned_data #for udpate if data['id']: emp = Employer.objects.filter(id = data['id']) if emp: emp = emp[0] emp.fname = data['fname'] emp.mname = data['mname'] emp.lname = data['lname'] emp.city = data['city'] emp.state = data['state'] emp.country = data['country'] emp.pincode = data['pincode'] emp.email = data['email'] emp.mobile = data['mobile'] emp.save() return emp else: #For direct save emp = Employer(fname=data['fname'], mname=data['mname'], lname=data['lname'], city=data['city'], state=data['state'], country=data['country'], pincode=data['pincode'], email=data['email'], mobile=data['mobile']) emp.save() return emp
class ModelTestCase(TestCase):
"""Define correct model creation."""
def setUp(self):
"""Add initial data."""
self.name = 'J'
self.new_employer = Employer(name=self.name)
def test_new_model_creation(self):
"""Check new model instantiation."""
initial_count = Employer.objects.count()
self.new_employer.save()
new_count = Employer.objects.count()
self.assertNotEqual(initial_count, new_count)
self.name2 = 'employe223'
self.new_employee = Employee(name=self.name2,
employer=self.new_employer)
self.new_employee.save()
self.assertEqual(len(Employee.objects.all()), 1)
def dict_to_employer(d):
"""
Convert dictionary to Employer object
:param d: dictionary
:return: Employer object
"""
em_obj = Employer()
em_obj.company_name = d['company_name'] if 'company_name' in d else ''
em_obj.description = d['description'] if 'description' in d else ''
em_obj.num_employees = d['num_employees'] if 'num_employees' in d else 0
em_obj.username = d['email']
em_obj.email = d['email']
em_obj.password = d['password']
em_obj.location = d['location'] if 'location' in d else ''
return em_obj
def add_rp14_form(dictionary):
with contextlib.closing(make_session()) as session:
employer = Employer()
employer.ip_number = dictionary["ip_number"]
employer.employer_name = dictionary["employer_name"]
employer.company_number = dictionary["company_number"]
employer.date_of_insolvency = dictionary["date_of_insolvency"]
employer.hstore = json_encode(dictionary)
session.add(employer)
session.commit()
def employer_profile(employer_id):
employer_info = map(lambda e: e.to_dict(),
Employer.objects(id=employer_id))
jobs = map(lambda job: job.to_dict(), Job.objects(employer_id=employer_id))
print employer_info
print jobs
return render_template('employer_profile.html',
employer_id=employer_id,
employer_info=employer_info,
jobs=jobs)
def done(self, form_list, form_dict, **kwargs):
with transaction.atomic():
user_data = form_dict['user'].cleaned_data
phone_number = user_data['country_code'] + user_data['phone_number']
email = user_data['email']
username = user_data['username']
user = form_dict['user'].save()
salt = hashlib.sha1(str(random.random())).hexdigest()[:5]
activation_key = hashlib.sha1(salt+email).hexdigest()
key_expires = timezone.now() + timedelta(days=30)
user_profile = UserProfile(user=user, phone=phone_number,
role='employer',
activation_key=activation_key,
key_expires=key_expires)
user_profile.save()
email_subject = 'Account Confirmation'
confirmation_url = self.request.build_absolute_uri(reverse('email_confirm',
kwargs={'activation_key': activation_key}))
email_body = "Hello %s, thanks for signing up. To activate your account,\
click this link %s" % (username, confirmation_url)
send_email.delay(email_subject, email_body, [email])
profile_data = form_dict['company_profile'].cleaned_data
employer = Employer(user=user, company=profile_data['company_name'],
description=profile_data['description'],
location=profile_data['location'],
address=profile_data['address'])
employer.save()
messages.success(self.request,
"You've successfully signed up. Please click the activation link sent to your email to activate your account")
return HttpResponseRedirect('/')
def to_dict(self):
_dict = {
'job_id' : str(self.id),
'employer_id' : str(self.employer_id),
'position' : self.position,
'description' : self.description,
'location' : self.location,
'openings' : self.openings,
'applications' : map(lambda app: app.to_dict(), Application.objects(job_id=self.id))
}
employer = Employer.by_id(self.employer_id)
if employer:
_dict.update({
'company_name' : employer.company_name
})
return _dict
def add_rp14_form(dictionary):
with contextlib.closing(make_session()) as session:
employer = Employer()
employer.ip_number = dictionary["ip_number"]
employer.employer_name = dictionary["employer_name"]
employer.company_number = dictionary["company_number"]
employer.date_of_insolvency = dictionary["date_of_insolvency"]
employer.hstore = {key: json.dumps(value, default=encode_special_types)
for key, value in dictionary.items()}
session.add(employer)
session.commit()
def before_request_handler():
if 'employer_id' in session:
employers = Employer.objects(id=session['employer_id'])
for employer in employers:
g.user = employer
g.employer = True
g.student = False
elif 'student_id' in session:
students = Student.objects(id=session['student_id'])
for student in students:
g.user = student
g.student = True
g.employer = False
else:
g.employer = False
g.student = False
g.user = False
def setUp(self):
"""Add initial data."""
self.name = 'J'
self.new_employer = Employer(name=self.name)
def create(self, validated_data):
user = None
employer = None
try:
user = User.objects.create_user(
email=validated_data.get('email'),
password=validated_data.get('password1'),
first_name=validated_data.get('first_name'),
last_name=validated_data.get('last_name'),
type_profile="employer")
employer = Employer()
employer.user = user
employer.gender = validated_data.get('gender')
employer.phone_number = validated_data.get('phone_number')
employer.mobile_number = validated_data.get('mobile_number')
employer.position = validated_data.get('position')
employer.name_company = validated_data.get('name_company')
employer.razon_social = validated_data.get('razon_social')
employer.responsability_iva = validated_data.get(
'responsability_iva')
employer.address = validated_data.get('address')
employer.city_id = validated_data.get('city', None)
employer.region_id = validated_data.get('region')
employer.country_id = validated_data.get('country')
employer.sector_empresarial = validated_data.get(
'sector_empresarial')
employer.workforce = validated_data.get('workforce')
employer.tipology = validated_data.get('tipology')
employer.description = validated_data.get('description')
employer.website = validated_data.get('website')
employer.cuit = validated_data.get('cuit')
# employer.logo = validated_data.get('logo')
employer.reference = validated_data.get('reference')
employer.newsletter = validated_data.get('newsletter')
employer.cv_spontany = validated_data.get('cv_spontany')
employer.save()
user.send_validation_email()
except Exception, e:
if employer:
try:
employer.delete()
except:
pass
if user:
user.delete()
raise e
