Allow overriding token name.
authorCMDR furrycat <elite@furrycat.net>
Mon, 26 Sep 2016 08:53:12 +0000 (09:53 +0100)
committerCMDR furrycat <elite@furrycat.net>
Mon, 26 Sep 2016 08:53:12 +0000 (09:53 +0100)
bot.py

diff --git a/bot.py b/bot.py
index e144406..deba921 100755 (executable)
--- a/bot.py
+++ b/bot.py
@@ -16,6 +16,10 @@ if sys.version_info >= (3, 0):
 else:
   import urllib2
 
+token_file = 'TOKEN'
+if len(sys.argv) > 1:
+  token_file = sys.argv[1]
+
 admins = []
 admin_role = 'Cat'
 fd = open('ADMINS', 'r')
@@ -24,7 +28,7 @@ for line in fd.readlines():
   admins.append(admin)
 fd.close()
 
-fd = open('TOKEN', 'r')
+fd = open(token_file, 'r')
 token = fd.readline().strip()
 fd.close()
 logging.basicConfig(level = logging.WARN)