diff options
Diffstat (limited to 'titlebot/commands')
| -rwxr-xr-x | titlebot/commands/undo | 10 | ||||
| -rwxr-xr-x | titlebot/commands/up | 18 | 
2 files changed, 19 insertions, 9 deletions
| diff --git a/titlebot/commands/undo b/titlebot/commands/undo index bebd57a..a66de67 100755 --- a/titlebot/commands/undo +++ b/titlebot/commands/undo @@ -16,12 +16,16 @@ except:      undos vote of one or more entries based on .list""")      sys.exit(1)  voter = environ['_prefix'] +voter_name = environ['_from']  for vote in votes:      try:          vote = int(vote) -        del(db[vote]['votes'][voter] ) -        print("undid vote by %s for #%d" %(environ['_from'],vote)) +        if not voter in db[vote]['votes']: +            print("%s, you never voted for '%s'!"%(voter_name,db[vote]['title'])) +        else: +            del(db[vote]['votes'][voter] ) +            print("%s undid vote for '%s'" %(voter_name,db[vote]['title'] ))      except: -        print("undo voting for #%s failed" %vote) +        print("%s undo voting for #%d failed" %(voter_name,vote))  poll.save_db(f,db) diff --git a/titlebot/commands/up b/titlebot/commands/up index c9de122..0a48bdb 100755 --- a/titlebot/commands/up +++ b/titlebot/commands/up @@ -9,19 +9,25 @@ import poll  f = 'suggestions.json'  db = poll.load_db(f)  votes = [] -try: -    votes = sys.argv[1:] -except: +votes = sys.argv[1:] +if not votes:      print("""usage: up number (...)      upvotes one or more entries based on .list""")      sys.exit(1)  voter = environ['_prefix'] +voter_name =environ['_from']  for vote in votes:      try:          vote = int(vote) -        db[vote]['votes'][voter] = 1 +        if vote < 0: +            raise Exception() +        if voter in db[vote]['votes']: +            print("%s, you already have voted for '%s'"%(voter_name,db[vote]['title']) ) +        else: +            db[vote]['votes'][voter] = 1 +            print("%s voted for '%s'"%(voter_name,db[vote]['title']))      except: -        print("voting for #%s failed" %vote) -print("Thanks for your votes %s"%environ['_from']) +        print("%s, voting for #%s failed" %(voter_name,vote)) +  poll.save_db(f,db) | 
