def generate_users(request):
if request.method == "POST":
# 量を取得
quantity = int(request.values.get("quantity", 100))
# 開始
log = Log()
log.name = u"generate users:%d" % (quantity,)
log.started_at = datetime.now()
# ランダムに指定数分先生を作る
users = []
for i in range(0, quantity):
user = User(name=generate_string(20), shift=random.choice([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]))
users.append(user)
# 何件ごとに書き込むか検討
if i % 20 == 0:
db.put(users)
users = []
if len(users) > 0:
db.put(users)
# 終了
log.ended_at = datetime.now()
log.put()
return Response(status=200)
def generate_schedules(request):
if request.method == "POST":
# 開始
log = Log()
log.name = u"generate schedules"
log.started_at = datetime.now()
# ユーザ全員を取得
users = User.all().fetch(1000)
# 日付を取得
today = datetime.today()
target_day = date(today.year, today.month + 1, 1)
nn_month = date(today.year, today.month + 2, 1)
# 1日ずつスケジュールを入れる
schedules = []
while target_day < nn_month:
# 時間をずらす
for hour in range(9, 24):
# 分をずらす
for minutes in [0, 30]:
# ユーザずつ
for user in users:
# シフト
# 0/1: 月曜休みAM/月曜休みPM
# 2/3: 火曜休みAM/火曜休みPM
weekday = target_day.weekday()
if user.shift == weekday * 2 or user.shift == weekday * 2 + 1:
continue
# AM: 9 - 17
# PM: 15 - 23
if user.shift % 2 == 0 and hour > 18:
continue
if user.shift % 2 == 1 and hour < 15:
continue
# スケジュール入れる
target_datetime = datetime(target_day.year, target_day.month, target_day.day, hour, minutes)
schedule = Schedule(
name=generate_string(20), user=user, title=generate_string(100), schedule_at=target_datetime
)
# リストに追加
schedules.append(schedule)
# 追加
db.put(schedules)
schedules = []
# 1日追加
target_day = target_day + timedelta(days=1)
# 終了
log.ended_at = datetime.now()
log.put()
return Response(status=200)