Added helper script to give ranks/badges

2017-11-09 16:31:34 +01:00 · 2017-11-09 16:31:34 +01:00 · 1725f0b519
parent a3c853e490
commit 1725f0b519
1 changed files with 111 additions and 0 deletions
--- a/give_honors_to_org.py
+++ b/give_honors_to_org.py
@ -0,0 +1,111 @@
+#!/usr/bin/env python3.5
+
+import os, sys
+import redis
+import configparser
+
+import contributor_helper
+
+configfile = os.path.join(os.environ['DASH_CONFIG'], 'config.cfg')
+cfg = configparser.ConfigParser()
+cfg.read(configfile)
+serv_redis_db = redis.StrictRedis(
+        host=cfg.get('RedisGlobal', 'host'),
+        port=cfg.getint('RedisGlobal', 'port'),
+        db=cfg.getint('RedisDB', 'db'))
+
+chelper = contributor_helper.Contributor_helper(serv_redis_db, cfg)
+
+def printOrgInfo(org):
+    org_pnts = chelper.getOrgContributionTotalPoints(org)
+    org_c_rank = chelper.getOrgContributionRank(org)
+    org_c_status = chelper.getCurrentContributionStatus(org)
+    org_honor_badge = chelper.getOrgHonorBadges(org)
+
+    print()
+    print("Organisation points: {}".format(org_pnts))
+    print("Organisation contribution rank: {}".format(org_c_status['rank']))
+    print('''
+Organisation contribution rank:
+-------------------------------''')
+    for rank in range(1, chelper.org_rank_maxLevel+1):
+        acq = 'x' if org_c_status['status'][rank] == 1 else ' '
+        print("{}.\t[{}]\t{}\t{}".format(rank, acq, chelper.org_rank_requirement_pnts[rank], chelper.org_rank_requirement_text[rank]))
+
+    print()
+    print('''
+Organisation honor badges:
+--------------------------''')
+    for badgeNum, text in chelper.org_honor_badge_title.items():
+        acq = 'x' if badgeNum in org_honor_badge else ' '
+        print("{}.\t[{}]\t{}".format(badgeNum, acq, text))
+    print()
+
+
+def main():
+    if len(sys.argv) > 1:
+        org = sys.argv[1]
+    else:
+        org = input('Enter the organisation name: ')
+    
+    printOrgInfo(org)
+
+    # ranks
+    while True:
+        org_pnts = chelper.getOrgContributionTotalPoints(org)
+        org_c_rank = chelper.getOrgContributionRank(org)
+        org_c_status = chelper.getCurrentContributionStatus(org)
+        org_honor_badge = chelper.getOrgHonorBadges(org)
+
+        userRep = input("Enter the organisation RANK to give/remove to {} (<ENTER> to finish): ".format(org))
+        if userRep == '':
+            break
+        else:
+            # validate input
+            try: #not int
+                rankNum = int(userRep)
+            except:
+                print('Not an integer')
+                continue
+            if rankNum < 1 and rankNum > chelper.org_rank_maxLevel:
+                print('Not a valid rank')
+                continue
+
+            if org_c_status['status'][rankNum] == 1: #remove rank
+                chelper.removeContribRankFromOrg(org, rankNum)
+            else:
+                chelper.giveContribRankToOrg(org, rankNum)
+
+            printOrgInfo(org)
+    
+    # badges
+    while True:
+        org_pnts = chelper.getOrgContributionTotalPoints(org)
+        org_c_rank = chelper.getOrgContributionRank(org)
+        org_c_status = chelper.getCurrentContributionStatus(org)
+        org_honor_badge = chelper.getOrgHonorBadges(org)
+
+        userRep = input("Enter the organisation BADGE to give/remove to {} (<ENTER> to finish): ".format(org))
+        if userRep == '':
+            break
+        else:
+            # validate input
+            try: #not int
+                badgeNum = int(userRep)
+            except:
+                print('Not an integer')
+                continue
+            if badgeNum < 1 and badgeNum > chelper.honorBadgeNum:
+                print('Not a valid rank')
+                continue
+
+            if badgeNum in org_honor_badge: #remove badge
+                chelper.removeBadgeFromOrg(org, badgeNum)
+            else:
+                chelper.giveBadgeToOrg(org, badgeNum)
+
+            printOrgInfo(org)
+
+if __name__ == '__main__':
+    main()
+