Adding howtos now works

git-svn-id: https://svn.lusy.fri.uni-lj.si/kpov-public-svn/kpov-public@190 5cf9fbd1-b2bc-434c-b4b7-e852f4f63414

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)