FilesExpand file tree

 master

/

create_data.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

45 lines (33 loc) · 1.58 KB

 master

/

create_data.py

Copy path

Top

File metadata and controls

• Code
• Blame

45 lines (33 loc) · 1.58 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

import csv

from datetime import date, timedelta

import random

from models import Company

def get_project_name():

projects = ['Ребрендинг', 'Разработка CRM', 'Обслуживание 1С', 'Разработка сайта',

'Опрос покупателей', 'Запуск колцентра', 'Модернизация wifi-сети',

'Проведение исследований', 'Дизайн сайта', 'Разработка моб. приложения',

'Дизайн буклетов', 'Аудит информационной безопасности',

'Обучение сотрудников']

return random.choice(projects)

def projects_for_employee(company_id, employee_id):

projects = []

for month in range(1, 13):

date_start = date(2020, month, random.randint(1, 10))

date_end = date_start + timedelta(days=random.randint(5, 20))

project = [get_project_name(), company_id, employee_id, date_start, date_end]

projects.append(project)

return projects

def fake_project_list():

data = []

companies = Company.query.all()

for company in companies:

for employee in company.employees:

data += projects_for_employee(company.id, employee.id)

return data

def generate_data(data, filename):

with open(filename, 'w', encoding='utf-8') as f:

writer = csv.writer(f, delimiter=';')

for row in data:

writer.writerow(row)

if __name__ == '__main__':

generate_data(fake_project_list(), 'projects.csv')