diff --git a/bot.py b/bot.py index ddb1804..5039ea2 100644 --- a/bot.py +++ b/bot.py @@ -2,6 +2,7 @@ # modules import os +import sys import time import json from pprint import pprint @@ -153,11 +154,11 @@ def backup_db(): 'status': user._status, 'question': user._question } - with open('users.json', 'w') as f: + with open(sys.path[0] + '/users.json', 'w') as f: json.dump(users_backup, f, indent=' ') def restore_db(): - with open('users.json', 'r') as f: + with open(sys.path[0] + '/users.json', 'r') as f: users_restore = json.load(f) for key in users_restore.keys(): user = users_restore[key] diff --git a/interface.py b/interface.py index edebb6a..28a0f37 100644 --- a/interface.py +++ b/interface.py @@ -2,6 +2,7 @@ # modules import os +import sys from random import choice import telepot from telepot.namedtuple import ReplyKeyboardMarkup, ReplyKeyboardRemove @@ -228,7 +229,7 @@ class Kuser: def send_answer(self, text, file_id): self._status = '正常使用' # define filename - filename = self._username + self._question + filename = sys.path[0] + '/' + self._username + self._question if self._api.list_questions()[self._question][3] == 'Python': filename += '.py' else: