def test_single_tag_multiple_users(self):
user1 = UserF(username="******")
person1 = Person(user=user1)
person1.save()
user2 = UserF(username="******")
person2 = Person(user=user2)
person2.save()
tag_category = TagCategory(name='Test Category',
slug='staff-directory-test-category')
tag_category.save()
tags = []
tagged_item = add_tags(person1, 'TagA',
'staff-directory-test-category', user1,
'person')
tagged_item = add_tags(person2, 'TagA',
'staff-directory-test-category', user1,
'person')
tags.append(tagged_item.tag.slug)
emails = _get_emails_for_tag(tags)
self.assertTrue(len(emails) == 2)
self.assertIn(user1.email, emails)
self.assertIn(user2.email, emails)
def test_normalized_phone_number(self):
from core.models import Person
p = Person(phone='0505551234')
self.assertEqual(p.normalized_phone_number, '+358 50 5551234')
p = Person(phone='ööää')
self.assertEqual(p.normalized_phone_number, 'ööää')
def add_user(req):
user = req.user
stub = user.username.split('@')[0]
user.email = req.POST['email']
user.first_name = req.POST['first_name']
user.last_name = req.POST['last_name']
user.save()
org_group = OrgGroup(pk=req.POST['team'])
office_location = OfficeLocation(pk=req.POST['office_location'])
attributes = {
'user': user,
'stub': stub,
'title': req.POST['title'],
'office_location': office_location,
'org_group': org_group,
'office_phone': req.POST['office_phone'],
'mobile_phone': req.POST['mobile_phone']
}
if req.FILES:
attributes['photo_file'] = req.FILES['photo_file']
person = Person(**attributes)
person.save()
return user
def _create_people(self):
self.people = []
for i in range(1, 4):
person_name = 'Person ' + str(i)
person = Person(name=person_name, rate=i)
person.save()
self.people.append(person)
def test_people_str(self):
"""Tests the validity of the people model"""
with open("core/tests/test_data/person.json") as person_json:
person = json.load(person_json)
person_obj = Person(**person)
person_obj.save()
assert Person.objects.get(
index=person['index']).name == person['name']
def test_get_absolute_uri(self):
person = Person(full_name='John Doe')
person.save()
self.assertEqual(
person.get_absolute_url(),
reverse('management:collaborator-detail', kwargs={'pk':
person.pk}))
def test_generating_slug(self):
person = Person(name='Charles Babbage')
person.save()
self.assertEqual(
person.slug,
'charles-babbage',
)
def save(self, request):
adapter = get_adapter()
user = adapter.new_user(request.data)
user.person_id = Person(person_id=request.data['person_id'])
self.cleaned_data = self.get_cleaned_data()
adapter.save_user(request, user, self)
user.set_password(request.data['password1'])
user.save()
return user
def signup(self, request, user): user.first_name = self.cleaned_data['first_name'] user.last_name = self.cleaned_data['last_name'] user.save() person = Person() person.role = self.cleaned_data['cargo'] person.user = user person.save()
def create_superuser():
user = get_user_model().objects.create_superuser('*****@*****.**',
'*****@*****.**', '1')
# If using collab, person objects needs to be created too
if 'core' in settings.INSTALLED_APPS:
from core.models import Person
person = Person()
person.user = user
person.save()
def test_money_context_for_person_processing(self):
ctx = getcontext()
ctx.prec = 1
ctx.rounding = ROUND_CEILING
self.assertEqual(ctx.prec, 1)
self.assertEqual(ctx.rounding, ROUND_CEILING)
_ = Person()
ctx = getcontext()
self.assertNotEqual(ctx.prec, 1)
self.assertNotEqual(ctx.rounding, ROUND_CEILING)
def get_user(self, user, name, profile_picture):
person = None
try:
person = Person.objects.get(login=user)
except Person.DoesNotExist:
person = Person(login=user,
name=name,
profile_picture=profile_picture)
person.save()
return person
def test_helper_no_profile(self):
u = User()
u.first_name = "John"
u.last_name = "Smith"
u.save()
self.assertEquals(False, helpers.user_has_profile(u))
p = Person()
p.user = u
p.save()
self.assertEquals(True, helpers.user_has_profile(u))
def create(self, validated_data):
data = {
'name': validated_data['name'],
'title': validated_data['title'],
}
label = validated_data.get('label', None)
student_id = validated_data.get('student_id', None)
if label is not None:
data['pk'] = label
if student_id is not None:
data['student_id'] = student_id
return Person(**data)
def test_create_tags_for_person(self):
user = UserF(username="******")
person = Person(user=user)
person.save()
tag_category = TagCategory(name='Test Category',
slug='staff-directory-test-category')
tag_category.save()
tagged_item = add_tags(person, 'TagA', 'staff-directory-test-category',
user, 'person')
self.assertEqual(1, person.tags.count())
def create_worker(data, party):
worker = Person()
worker.party = party
worker.firstName = data.get('firstName')
worker.secondName = data.get('secondName')
worker.patronymicName = data.get('patronymicName')
worker.birthDate = data.get('birthDate')
worker.startDate = datetime.now()
worker.endDate = date.max
worker.gender = data.get('gender')
worker.infoText = data.get('infoText')
worker.save()
return worker
def setUp(self):
super().setUp()
self.client = Client(schema)
# Populate the test collectiom with person data
with open("core/tests/test_data/employees.json") as employee_json:
employees = json.load(employee_json)
for employee in employees:
employee_obj = Person(**employee)
employee_obj.save()
# Populate the test collection with company data
with open("core/tests/test_data/companies.json") as companies_json:
companies = json.load(companies_json)
for companies in companies:
companies_obj = Company(**companies)
companies_obj.save()
def fill_db():
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '1'
person.account = '1'
person.save()
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '2'
person.account = '0'
person.save()
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '123'
person.account = '50'
person.save()
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '456'
person.account = '50'
person.save()
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '452345134234'
person.account = '-100'
person.save()
user = User()
user.username = '******'
user.save()
person = Person()
person.user = user
person.inn = '67896'
person.account = '0.99'
person.save()
def test_single_tag_single_user(self):
user = UserF(username="******")
person = Person(user=user)
person.save()
tag_category = TagCategory(name='Test Category',
slug='staff-directory-test-category')
tag_category.save()
tags = []
tagged_item = add_tags(person, 'TagA', 'staff-directory-test-category',
user, 'person')
tags.append(tagged_item.tag.slug)
emails = _get_emails_for_tag(tags)
self.assertTrue(len(emails) == 1)
self.assertIn(user.email, emails)
def setUp(self):
super().setUp()
self.client = Client(schema)
# Populate the test collectiom with person data
with open("core/tests/test_data/people.json") as people_json:
peoples = json.load(people_json)
for people in peoples:
people_obj = Person(**people)
people_obj.save()
# Populate the food collection from test data file
with open("core/tests/test_data/food.json") as food_json:
food_list = json.load(food_json)
for food in food_list:
food_obj = Food(**food)
food_obj.save()
def test_single_tag_user_profile(self):
user = UserF(username="******")
user.set_password("x")
user.save()
person = Person(user=user)
person.stub = "jack"
person.save()
resp1 = self.client.login(username="******", password="******")
self.assertTrue(resp1)
tag_category = TagCategory(name='Test Category',
slug='staff-directory-test-category')
tag_category.save()
resp2 = self.client.post(
'/staff/add-person-to-tag/%s/hello!/' % (person.stub), {
'person_stub': person.stub,
'tag_category': 'staff-directory-test-category'
})
self.assertContains(resp2, "redirect", status_code=200)
def setUp(self):
# create a load of test people in the database
names = [
'Adam Ant',
'Bobby Smith',
'Fred Jones',
'Joe Bloggs',
'Joe Smith',
'Josepth Smyth',
]
for name in names:
Person(
slug=slugify(name),
legal_name=name,
gender='m',
).save()
# Haystack indexes are not touched when fixtures are dumped. Run this
# so that other changes cannot affect these tests. Have added a note to
# a HayStack issue regarding this:
# https://github.com/toastdriven/django-haystack/issues/226
call_command('rebuild_index', interactive=False)
def handle(self, *args, **kwargs):
for first_name in self.first_names:
for last_name in self.last_names:
person = Person(first_name=first_name, last_name=last_name)
person.save()
for producer_name in self.producer_names:
prod = Producer(address=choice(self.street_names) + ' ' +
str(randint(1, 200)) + '-' + str(randint(1, 500)),
name=producer_name)
prod.save()
for taste in self.tastes:
liquid = Liquid(producer=choice(Producer.objects.all()),
price=Decimal(randrange(10, 100) / 10),
taste=taste)
liquid.save()
for _ in range(100):
tab = Tab(producer=choice(Producer.objects.all()),
price=Decimal(randrange(20, 300) / 10),
power=randint(50, 300))
tab.save()
for _ in range(70):
evaporator = Evaporator(producer=choice(Producer.objects.all()),
price=Decimal(randrange(50, 600) / 10))
evaporator.save()
def test_days_since_hire(self):
person = Person(user=User(first_name="Baba", last_name="O'Reilly"))
person.start_date = timezone.now() - datetime.timedelta(days=10)
self.assertEquals(person.days_since_hire, 10)
def subscribe_service(request, key):
service = Service.get(key)
form = SubscribeForm()
person_email_content = ''
if request.method == 'POST':
form = SubscribeForm(request.POST)
if form.is_valid():
first_name = form.cleaned_data['first_name']
last_name = form.cleaned_data['last_name']
email = form.cleaned_data['email']
type_email = form.cleaned_data['type_email']
if service.person_mail == True:
person_email_content = request.POST['person_mail']
person = Person.all().filter("person_email =", email).get()
if person == None:
person = Person(person_first_name=first_name,
person_last_name=last_name,
person_email=email,
person_is_activ=True,
person_type_email=type_email)
person.put()
else:
person.person_is_activ = True
person.put()
subscr_query = Subscribe.all().filter("subscribe_user ="******""
for dest in service.responsable:
user = get_object(User, dest)
to += render_to_string('service/subscribe_email_to.txt',
{'user': user})
message.to = to
message.send()
return render_to_response(request, 'service/subscribe_ok.html',
{'service': service})
return render_to_response(request, 'service/subscribe_form.html', {
'service': service,
'form': form
})
def respond_with_new_user(self, request, next_url, desuprofile):
"""
This implements the following case:
2. No Kompassi account is linked to this Desuprofile, and no Kompassi account matches the
Desuprofile by email address. A new Kompassi account is created and logged in.
"""
User = get_user_model()
password = create_temporary_password()
# Kompassi has stricter rules for username validation than Desusite
username = desuprofile.username.lower()
try:
valid_username(username)
except DjangoValidationError:
username = None
else:
try:
User.objects.get(username=username)
except User.DoesNotExist:
# Username is free
pass
else:
# Username clash with an existing account, use safe username
username = None
if username is None:
username = "******".format(id=desuprofile.id)
with transaction.atomic():
user = User(
username=username,
is_active=True,
is_staff=False,
is_superuser=False,
)
user.set_password(password)
user.save()
person = Person(
first_name=desuprofile.first_name.strip(),
surname=desuprofile.last_name.strip(),
nick=desuprofile.nickname.strip(),
email=desuprofile.email.strip(),
phone=desuprofile.phone.strip(),
birth_date=(
datetime.strptime(desuprofile.birth_date.strip(), '%Y-%m-%d').date()
if desuprofile.birth_date
else None
),
notes='Luotu Desuprofiilista',
user=user,
)
person.save()
connection = Connection(
id=int(desuprofile.id),
desuprofile_username=desuprofile.username,
user=user,
)
connection.save()
person.apply_state_new_user(request, password)
messages.success(request, 'Sinulle on luotu Desuprofiiliisi liitetty Kompassi-tunnus. Tervetuloa Kompassiin!')
return respond_with_connection(request, next_url, connection)
def handle(self, *args, **options):
last_names = [
'Sherman', 'Maddox', 'Montgomery', 'Small', 'Larsen', 'Marsh',
'Gardner', 'White', 'Gill', 'Pennington', 'Stein', 'Kirby',
'Jennings', 'French', 'Glass', 'Velasquez', 'Doyle', 'York',
'Fisher', 'Strong', 'Henson', 'Harmon', 'Higgins', 'James',
'Hancock', 'Drake', 'Eaton', 'Gordon', 'Harrington', 'Blevins',
'Avila', 'Solis', 'Richmond', 'Stark', 'Haynes', 'Durham',
'Montoya', 'Barrett', 'Chase', 'Mckay', 'Little', 'Perry',
'Howard', 'Caldwell', 'West', 'Fox', 'Long', 'Wright', 'Foster',
'Sloan', 'Frazier', 'Lowe', 'Cabrera', 'Barron', 'Ayala', 'Frank',
'Hammond', 'Orr', 'Holloway', 'King', 'Rush', 'Wiley', 'Neal',
'Davis', 'Fulton', 'Webb', 'Sanchez', 'Strickland', 'Clark',
'Middleton', 'Moody', 'Owens', 'Graham', 'Cotton', 'Shaffer',
'Hawkins', 'Cooper', 'Justice', 'Clarke', 'Mcconnell', 'Mccarthy',
'Macdonald', 'Castillo', 'Gilbert', 'Horton', 'Finley', 'Beard',
'Sanders', 'Levy', 'Richard', 'Bowen', 'Grant', 'Wilkins',
'Ramsey', 'Lynch', 'Koch', 'Mercado', 'Roach', 'Bond', 'Lane',
'Tanner', 'Byers', 'Humphrey', 'Austin', 'Carney', 'Golden',
'Pope', 'Kramer', 'Ellison', 'Jefferson', 'Duffy', 'Gross',
'Mcmahon', 'Hudson', 'Mckee', 'Atkinson', 'Bush', 'Thompson',
'Faulkner', 'Christian', 'Ingram', 'Cannon', 'Gay', 'Nieves',
'Hodges', 'Langley', 'Watson', 'Woods', 'Gallagher', 'Delacruz',
'Stafford', 'Knight', 'Kerr', 'Chapman', 'Roman', 'Christensen',
'Robles', 'Mathews', 'Waller', 'Buckley', 'Myers', 'Powers',
'Lindsay', 'Gates', 'Miller', 'Johns', 'Morin', 'Fleming',
'Bishop', 'Clements'
]
first_names = [
'Sarah', 'Ian', 'Upton', 'Uriah', 'Hayden', 'Zia', 'Lila',
'Benjamin', 'Addison', 'Vivian', 'Kirby', 'Oscar', 'Demetrius',
'Hashim', 'Michelle', 'Odessa', 'Phillip', 'Michael', 'Dante',
'Omar', 'Dominic', 'Wing', 'Joshua', 'Charlotte', 'Thomas',
'Aquila', 'Rana', 'Jolene', 'Felix', 'Cailin', 'Tatiana', 'Oprah',
'Belle', 'Sydnee', 'Kuame', 'Fleur', 'Matthew', 'Sylvia', 'Mary',
'Deborah', 'Ross', 'Hyacinth', 'Jacqueline', 'Jessica', 'Callie',
'Ariana', 'Leo', 'Desiree', 'Lunea', 'Chava', 'Jorden', 'Rudyard',
'Cally', 'Knox', 'Arthur', 'Dana', 'Rebekah', 'Yen', 'Hadassah',
'Duncan', 'Ginger', 'Valentine', 'Ivana', 'Iona', 'Jemima',
'Dorothy', 'Joan', 'Timothy', 'Amity', 'Uriel', 'Skyler', 'Phelan',
'Alma', 'Hadley', 'Quemby', 'Sonya', 'Axel', 'Slade', 'Riley',
'Rajah', 'Giselle', 'Selma', 'Nadine', 'Pascale', 'Carol', 'Steel',
'Lane', 'Emi', 'Trevor', 'Wyatt', 'Claire', 'Harlan', 'Liberty',
'Alexandra', 'Avram', 'Barbara', 'Rashad', 'Holmes', 'Kenneth',
'Preston', 'Patience', 'Adele', 'Alfonso', 'Harrison', 'Julian',
'Jena', 'Peter', 'Kessie', 'Katell', 'Denton', 'Piper', 'Jerry',
'Teegan', 'Chandler', 'Walter', 'Cheryl', 'Desirae', 'Tasha',
'Hunter', 'Logan', 'Tatyana', 'Gail', 'Galvin', 'Dara', 'Athena',
'Kay', 'Dustin', 'Faith', 'Mariam', 'Leroy', 'Edan', 'Alexis',
'Nissim', 'Octavia', 'Kareem', 'Heidi', 'Aspen', 'Gregory',
'Garrison', 'Jolie', 'Gloria', 'Alec', 'Asher', 'Julie', 'Ayanna',
'Gavin', 'Germane', 'Bertha', 'Quinn', 'Tarik'
]
office_location = OfficeLocation.objects.get(pk='DC123')
for x in xrange(int(args[0])):
while True:
first_name = first_names[int(random.random() *
len(first_names))]
last_name = last_names[int(random.random() * len(last_names))]
username = last_name + first_name[0]
email = username + '@exmaple.com'
if not get_user_model().objects.filter(
username=email).exists():
break
user_attr = {
'first_name': first_name,
'last_name': last_name,
'is_active': True,
'is_superuser': False,
'date_joined': '2012-03-01T16:03:14Z',
'password':
'******',
'is_staff': True,
'email': email,
'username': email
}
user = get_user_model(**user_attr)
user.save()
person_attr = {
'office_location': office_location,
'allow_tagging': True,
'photo_file': 'avatars/default1.jpg',
'stub': username.replace('.', ''),
'office_phone': '5555555555',
'user': user,
'email_notifications': False,
'org_group': OrgGroup.objects.order_by('?')[0]
}
person = Person(**person_attr)
person.save()
self.stdout.write('Successfully created %d users and persons' %
int(args[0]))
def test_str(self):
person = Person(full_name='John Doe')
self.assertEqual(str(person), 'John Doe')
def handle(self, *args, **options):
peklun = User.objects.get(username="******")
wks = get_spreadsheet().sheet1
for i, l in enumerate(wks.get_all_records()):
# reopen it time from time to avoid disconnect by timeout
if i % 2000 == 0 and i:
wks = get_spreadsheet().sheet1
self.stdout.write('Processing line #{}'.format(i))
company_ipn = l.get("ІПН", "")
company_name = l.get("Назва", "")
person_id = l.get("id персони", "")
company_id = l.get("id компанії", "")
photo_url = l.get("Фото", "")
person = None
# First let's search for appropriate company
company = self.process_company(company_id, company_ipn,
company_name)
# No company — no go
if company is None:
continue
# Let's backwrite company id to the spreadsheet for further use
if company.pk != company_id:
company_id = company.pk
wks.update_cell(i + 2, len(l.keys()), company.pk)
person_name = l.get("ПІБ", "").strip()
position = l.get("Посада", "").strip()
person_dob = unicode(l.get("Дата народження", "")).strip()
person_from = parse_date(l.get("Дата призначення", ""))
person_to = parse_date(l.get("Дата звільнення", ""))
doc_received = parse_date(l.get("Дата відповіді", ""))
docs = l.get("Лінк на відповідь", "").strip()
website = l.get("лінк на сайт", "").strip()
# Now let's search for the person
if person_name:
last_name, first_name, patronymic, _ = parse_fullname(
person_name)
if not last_name:
continue
# First we search by person_id (if it's present)
if person_id:
try:
person = Person.objects.get(pk=person_id)
except Person.DoesNotExist:
pass
# If nothing is found we search for name (for now)
if not person:
try:
person = Person.objects.get(
first_name_uk__iexact=first_name,
last_name_uk__iexact=last_name,
patronymic_uk__iexact=patronymic)
except Person.MultipleObjectsReturned:
self.stderr.write(
"Double person {}!".format(person_name))
except Person.DoesNotExist:
pass
# If nothing is found, let's create a record for that person
if not person:
person = Person()
self.stderr.write(
"Created new person {}".format(person_name))
person.first_name_uk = first_name
person.last_name_uk = last_name
person.patronymic_uk = patronymic
Ua2RuDictionary.objects.get_or_create(term=first_name)
Ua2RuDictionary.objects.get_or_create(term=last_name)
Ua2RuDictionary.objects.get_or_create(term=patronymic)
person.first_name_en = translitua(first_name)
person.last_name_en = translitua(last_name)
person.patronymic_en = translitua(patronymic)
person.is_pep = True
person.imported = True
person.type_of_official = 1
# Parsing date (can be a full date or just a year or
# year/month)
if person_dob:
person.dob = parse_date(person_dob)
if len(person_dob) == 4:
person.dob_details = 2 # Only year
if len(person_dob) > 4 and len(person_dob) < 7:
person.dob_details = 1 # month and year
# Let's download the photo (if any)
if not person.photo and photo_url:
photo_name, photo_san_name, photo_content = download(
photo_url, translitua(person_name))
if photo_name:
person.photo.save(photo_san_name,
ContentFile(photo_content))
else:
self.stdout.write("Cannot download image %s for %s" %
(photo_url, person_name))
person.save()
# Let's write the person id back to the table.
if person.pk != person_id:
person_id = person.pk
wks.update_cell(i + 2, len(l.keys()) - 1, person.pk)
# Now let's download all supporting docs
docs_downloaded = []
first_doc_name = False
# There might be many of them
for doc in docs.split(", "):
doc_instance = None
# we cannot download folders from google docs, so let's
# skip them
if doc and "folderview" not in doc \
and "drive/#folders" not in doc:
doc = expand_gdrive_download_url(doc)
doc_hash = sha1(doc).hexdigest()
# Check, if docs
try:
doc_instance = Document.objects.get(hash=doc_hash)
except Document.DoesNotExist:
self.stdout.write(
'Downloading file {}'.format(doc))
doc_name, doc_san_name, doc_content = download(doc)
doc_san_name = translitua(doc_san_name)
if doc_name:
doc_instance = Document(name_uk=doc_name,
uploader=peklun,
hash=doc_hash)
doc_instance.doc.save(doc_san_name,
ContentFile(doc_content))
doc_instance.save()
else:
self.stdout.write(
'Cannot download file {}'.format(doc))
if doc_instance:
first_doc_name = doc_instance.name_uk
docs_downloaded.append(doc_instance.doc.url)
# Now let's setup links between person and companies
links = Person2Company.objects.filter(
(Q(date_established=person_from)
| Q(date_established=mangle_date(person_from))
| Q(date_established__isnull=True)),
(Q(date_finished=person_to)
| Q(date_finished=mangle_date(person_to))
| Q(date_finished__isnull=True)),
from_person=person,
to_company=company)
# Delete if there are doubling links
# including those cases when dates were imported incorrectly
# because of parse_date
if len(links) > 1:
links.delete()
link, _ = Person2Company.objects.update_or_create(
from_person=person,
to_company=company,
date_established=person_from,
date_established_details=0,
date_finished=person_to,
date_finished_details=0)
if not link.relationship_type:
link.relationship_type = position
# And translate them
Ua2EnDictionary.objects.get_or_create(
term=lookup_term(position))
# oh, and add links to supporting docs
all_docs = docs_downloaded + website.split(", ")
if all_docs:
link.proof = ", ".join(filter(None, all_docs))
if first_doc_name:
link.proof_title = first_doc_name
link.date_confirmed = doc_received
link.is_employee = True
link.save()
from core.models import Person
from core.serializers import PersonSerializer
from rest_framework.renderers import JSONRenderer
from rest_framework.parsers import JSONParser
person = Person(first_name='Paul',
last_name='Van Dyke',
email='*****@*****.**')
person.save()
person = Person(first_name='Regis',
last_name='Santos',
email='*****@*****.**')
person.save()
serializer = PersonSerializer(person)
serializer.data
content = JSONRenderer().render(serializer.data)
content
