From e22e8d5e335af7f41e1947aad1a514f71677e57e Mon Sep 17 00:00:00 2001
From: Mattia Rizzolo <mattia@mapreri.org>
Date: Mon, 4 May 2015 14:36:13 +0200
Subject: reproducible: remote_scheduler: use the new DB table. Limit every
 person to 50 scheduling per day. This is really easy to bypass, but we have
 trust in you.

---
 bin/reproducible_remote_scheduler.py | 38 ++++++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 6 deletions(-)

(limited to 'bin/reproducible_remote_scheduler.py')

diff --git a/bin/reproducible_remote_scheduler.py b/bin/reproducible_remote_scheduler.py
index fcd38561..0b387387 100755
--- a/bin/reproducible_remote_scheduler.py
+++ b/bin/reproducible_remote_scheduler.py
@@ -8,8 +8,8 @@
 #
 # A secure script to be called from remote hosts
 
+import time
 import argparse
-from subprocess import check_output
 
 
 parser = argparse.ArgumentParser(
@@ -97,6 +97,8 @@ message += ': ' + ' '.join(packages)[0:256] + blablabla + artifacts_txt
 # so schedule them in the past, so they are picked earlier :)
 # the current date is subtracted twice, so it sorts before early scheduling
 # schedule on the full hour so we can recognize them easily
+epoch = int(time.time())
+yesterday = epoch - 60*60*24
 now = datetime.datetime.now()
 days = int(now.strftime('%j'))*2
 hours = int(now.strftime('%H'))*2
@@ -105,18 +107,42 @@ time_delta = datetime.timedelta(days=days, hours=hours, minutes=minutes)
 date = (now - time_delta).strftime('%Y-%m-%d %H:%M')
 log.debug('date_scheduled = ' + date + ' time_delta = ' + str(time_delta))
 
+
+# a single person can't schedule more than 50 packages in the same day; this
+# is actually easy to bypass, but let's give some trust to the Debian people
+query = '''SELECT count(*) FROM manual_scheduler
+           WHERE requester = "{}" AND date_request > "{}"'''
+try:
+    amount = int(query_db(query.format(requester, date))[0][0])
+except IndexError:
+    amount = 0
+log.debug(requester + ' already scheduled ' + str(amount) + ' packages today')
+if amount + len(ids) > 50:
+    log.error(bcolors.FAIL + 'You exceeded the maximun amount of manual ' +
+              'rescheduling for today. Please ask in #debian-reproducible ' +
+              'if need to schedule more packages.' + bcolors.ENDC)
+    sys.exit(1)
+
+
+# do the actual scheduling
 to_schedule = []
+save_schedule = []
 for id in ids:
     artifacts_value = 1 if artifacts else 0
-    to_schedule.append((id, date, artifacts_value))
+    to_schedule.append((id, date, artifacts_value, requester))
+    save_schedule.append((id, requester, epoch))
 log.debug('Packages about to be scheduled: ' + str(to_schedule))
 
-query = '''REPLACE INTO schedule
-    (package_id, date_scheduled, date_build_started, save_artifacts, notify)
-    VALUES (?, ?, "", ?, "true")'''
+query1 = '''REPLACE INTO schedule
+    (package_id, date_scheduled, date_build_started, save_artifacts, notify, scheduler)
+    VALUES (?, ?, "", ?, "true", ?)'''
+query2 = '''INSTERT INTO manual_scheduler
+    (package_id, requester, date_request) VALUS (?, ?, ?)'''
 
 cursor = conn_db.cursor()
-cursor.executemany(query, to_schedule)
+cursor.executemany(query1, to_schedule)
+cursor.executemanu(query2, save_schedule)
+conn_db.commit()
 
 log.info(bcolors.GOOD + message + bcolors.ENDC)
 irc_msg(message)
-- 
cgit v1.2.3-70-g09d2