73 lines
3.4 KiB
Python
73 lines
3.4 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import numpy as np
|
|
from django.core.exceptions import ValidationError
|
|
from django.core.management import BaseCommand
|
|
|
|
from web.models import Activity, ActivityLog, User
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
User.objects.all().delete()
|
|
Activity.objects.all().delete()
|
|
ActivityLog.objects.all().delete()
|
|
|
|
activity_sleeping = Activity(type='Сон')
|
|
activity_sleeping.save()
|
|
activity_eating = Activity(type='Еда')
|
|
activity_eating.save()
|
|
activity_studying = Activity(type='Учёба')
|
|
activity_studying.save()
|
|
activity_working = Activity(type='Работа')
|
|
activity_working.save()
|
|
activity_logistics = Activity(type='Транспорт')
|
|
activity_logistics.save()
|
|
activity_pastime = Activity(type='Досуг')
|
|
activity_pastime.save()
|
|
|
|
user1 = User(name='Вася', email='vasya@abra.me')
|
|
user1.save()
|
|
user2 = User(name='Петя', email='petya@abra.me')
|
|
user2.save()
|
|
|
|
start_date = datetime(2018, 12, 1, 0, 0, 0)
|
|
days = (datetime.utcnow() - start_date) // timedelta(days=1) - 1
|
|
|
|
for i in range(days):
|
|
day_start = start_date + timedelta(days=1) * i
|
|
|
|
def add_activity(user, activity, start_offset, end_offset, variance):
|
|
activity_log = ActivityLog(
|
|
user=user,
|
|
activity=activity,
|
|
start_time=day_start + start_offset + variance * abs(np.random.normal()),
|
|
end_time=day_start + end_offset - variance * abs(np.random.normal()),
|
|
)
|
|
try:
|
|
activity_log.clean()
|
|
except ValidationError as e:
|
|
print(e.message)
|
|
return
|
|
|
|
activity_log.save()
|
|
print(activity_log)
|
|
|
|
add_activity(user1, activity_sleeping, timedelta(hours=-24 + 22), timedelta(hours=8), timedelta(hours=2))
|
|
add_activity(user1, activity_eating, timedelta(hours=8), timedelta(hours=8.5), timedelta(minutes=5))
|
|
add_activity(user1, activity_logistics, timedelta(hours=8.5), timedelta(hours=9.5), timedelta(minutes=10))
|
|
add_activity(user1, activity_working, timedelta(hours=9.5), timedelta(hours=14), timedelta(minutes=30))
|
|
add_activity(user1, activity_eating, timedelta(hours=14), timedelta(hours=14.5), timedelta(minutes=5))
|
|
add_activity(user1, activity_working, timedelta(hours=14.5), timedelta(hours=18), timedelta(minutes=30))
|
|
add_activity(user1, activity_logistics, timedelta(hours=18), timedelta(hours=19), timedelta(minutes=10))
|
|
add_activity(user1, activity_eating, timedelta(hours=20), timedelta(hours=20.5), timedelta(minutes=5))
|
|
add_activity(user1, activity_pastime, timedelta(hours=20.5), timedelta(hours=22), timedelta(minutes=10))
|
|
|
|
add_activity(user2, activity_sleeping, timedelta(hours=0), timedelta(hours=6), timedelta(hours=1))
|
|
add_activity(user2, activity_pastime, timedelta(hours=6), timedelta(hours=7), timedelta(minutes=10))
|
|
add_activity(user2, activity_logistics, timedelta(hours=7), timedelta(hours=8), timedelta(minutes=10))
|
|
add_activity(user2, activity_studying, timedelta(hours=8), timedelta(hours=18), timedelta(minutes=30))
|
|
add_activity(user2, activity_logistics, timedelta(hours=18), timedelta(hours=19), timedelta(minutes=10))
|
|
add_activity(user2, activity_eating, timedelta(hours=19), timedelta(hours=20), timedelta(minutes=10))
|
|
add_activity(user2, activity_pastime, timedelta(hours=20), timedelta(hours=24), timedelta(minutes=10))
|