def _save_authinfo_list(self):
authinfo_data = {}
for k, authinfo in self.authinfo_list.items():
userid, password = authinfo
authinfo_data[k] = (self.enc.encrypt(userid),
self.enc.encrypt(password))
save(authinfo_data, self.FILE_PATH)
def main():
from sys import argv
from time import time
from pickle_utils import pickle_open_and_write as save, pickle_list_with_newlines as save_list
argc = len(argv)
if argc <= len(CLI_ARGS):
print "Usage: %s" % argv[0], ' '.join(CLI_ARGS)
print "Currently missing arguments:", ' '.join(CLI_ARGS[len(argv)-1:])
exit()
global mdb_host
k = int(argv[1].strip())
mdb_host = argv[2].strip()
filter_queries_file = argv[3].strip()
allowed_users_file = argv[4].strip()
queries_ids_mapping_file = argv[5].strip()
output_path = argv[6].strip()
t_init = time()
result = top_k_queries(k, mdb_host, filter_queries_file, allowed_users_file, queries_ids_mapping_file)
print "Done in", time()-t_init
t0 = time()
save(output_path, result, save_list)
print "Done in", time()-t0
print "Script run in", time()-t_init
def rmmemo(self, bitslack_obj, event, args):
if self.is_bot(event):
return None
user_str = self.get_event_user_str(event)
if len(args) > 1 and isinstance(args, list):
key = args[0][0]
texts = u'%s メモ %s はありません。' % (user_str, key)
if key in self.memo_data:
del self.memo_data[key]
save(self.memo_data, self.file_path)
texts = u'%s メモ %s を削除しました。' % (user_str, key)
return texts
return u'引数が足りません。'
def rmmemo(self, bitslack_obj, event, args):
if self.is_bot(event):
return None
user_str = self.get_event_user_str(event)
if len(args) > 1 and isinstance(args, list):
key = args[0][0]
texts = u'%s メモ %s はありません。' % (user_str, key)
if key in self.memo_data:
del self.memo_data[key]
save(self.memo_data, self.file_path)
texts = u'%s メモ %s を削除しました。' % (user_str, key)
return texts
return u'引数が足りません。'
def mkmemo(self, bitslack_obj, event, args):
if self.is_bot(event):
return None
user_str = self.get_event_user_str(event)
if len(args[0][0]) > 1 and isinstance(args[0][0], tuple):
key = args[0][0][0]
value = args[0][0][1]
texts = u'%s メモ %s を更新しました。' % (user_str, key)
if key in self.memo_data:
texts = u'%s メモ %s を作成しました。' % (user_str, key)
self.memo_data[key] = value
save(self.memo_data, self.file_path)
return texts
return u'引数が足りません。'
def mkmemo(self, bitslack_obj, event, args):
if self.is_bot(event):
return None
user_str = self.get_event_user_str(event)
if len(args[0][0]) > 1 and isinstance(args[0][0], tuple):
key = args[0][0][0]
value = args[0][0][1]
texts = u'%s メモ %s を更新しました。' % (user_str, key)
if key in self.memo_data:
texts = u'%s メモ %s を作成しました。' % (user_str, key)
self.memo_data[key] = value
save(self.memo_data, self.file_path)
return texts
return u'引数が足りません。'
def _on_im_at(self, bitslack_obj, event, args):
help = self._help(bitslack_obj, event, args)
if help is not None:
return help
where = self._parse_where(event['text'], u'im_at')
texts = []
userid = u'<@%s>' % (event['user'])
whereis_data = load(self.file_path)
old_where = None
if userid in whereis_data:
old_where = whereis_data[userid]
whereis_data[userid] = where
save(whereis_data, self.file_path)
if old_where is not None:
texts.append(u'%s: 居場所を変更しました。(%s -> %s)' % (userid, old_where, where))
else:
texts.append(u'%s: 居場所を設定しました。(%s)' % (userid, where))
return texts
def _save_authinfo_list(self):
authinfo_data = {}
for k, authinfo in self.authinfo_list.items():
userid, password = authinfo
authinfo_data[k] = (self.enc.encrypt(userid), self.enc.encrypt(password))
save(authinfo_data, self.FILE_PATH)