def handle(self, *args, **kwargs): path = kwargs['path'] df = pd.read_csv(path) upper_range = len(df) print("Wait Data is being Loaded") try: proj = [ Project( program_id=Program.objects.get( code=int(df['Programme Code'][row])), name=(df['PROJECT/COMPONENT NAME'][row]).strip(), code=df['Project/Component Code'][row], ) for row in range(0, upper_range) ] proj_data = Project.objects.bulk_create(proj) if proj_data: self.stdout.write('Successfully loaded Partner data ..') # for row in range(0, upper_range): # print(df['Program Code'][row]) # print(Program.objects.get(code=int(df['Program Code'][row]))) except Exception as e: print(e)
def create(self, validated_data): instance = Project(**validated_data) instance.creator = self.context['request'].user organization = self.get_organization() instance.organization = organization instance.save() ProjectContributor.objects.create(user=instance.organization.contact, project=instance, project_admin=True) return instance
employees = ['Noah Smith', 'Olivia Black', 'Liam Lampros', 'Ava Campedelli'] for fullname in employees: first, last = fullname.split() employee = Employee() employee.user = User.objects.create_user(first_name=first, last_name=last, username=first.lower(), email='*****@*****.**' % first.lower(), password='******') employee.save() projects = ['Project A', 'Project B', 'Project C', 'Project D', 'Project E'] employees = Employee.objects.all() for project_name in projects: project = Project() project.name = project_name project.save() project.employees.add(*employees) # yesterday, today and tomorrow worked_days = [ datetime.date.today() - datetime.timedelta(days=1), datetime.date.today(), datetime.date.today() + datetime.timedelta(days=1) ] times = [0, 1, 2, 4, 6, 8] for date in worked_days: for project in Project.objects.all(): for employee in Employee.objects.all(): hours = random.choice(times)