diff options
Diffstat (limited to 'kpov_judge/add_task.py')
-rwxr-xr-x | kpov_judge/add_task.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/kpov_judge/add_task.py b/kpov_judge/add_task.py index ea7aec6..bf2588c 100755 --- a/kpov_judge/add_task.py +++ b/kpov_judge/add_task.py @@ -8,7 +8,6 @@ import sys import inspect import kpov_random_helpers import settings -import guestfs import glob import os @@ -89,9 +88,10 @@ if __name__ == '__main__': if howto_lang not in {'images'}: with open(os.path.join(howto_dir, 'index.html')) as f: db.howtos.update({'task_id': task_id, 'lang': howto_lang}, - {'$set': {'text': f.read()}}) + {'$set': {'text': f.read()}}, upsert=True) else: for img in glob.glob(os.path.join(howto_dir, '*')): + fname = os.path.basename(img) with open(img) as f: db.howto_images.update({'task_id': task_id, 'fname': fname}, - {'$set': {'data': Binary(f.read())}}) + {'$set': {'data': Binary(f.read())}}, upsert=True) |