Merge pull request #5 from youjying/patch-1

user nonexistance error
2014-03-21 14:29:09 +00:00 · 2014-03-21 14:29:09 +00:00 · 7e3019259c
commit 7e3019259c
parent 652dd310f2 a8a0b09be2
1 changed files with 7 additions and 0 deletions
--- a/mining/DBInterface.py
+++ b/mining/DBInterface.py
@ -179,6 +179,13 @@ class DBInterface():
                self.dbi.insert_worker(uid, username, password)
                self.cache.set(username, password)
                return True
+            else:
+                self.dbi.insert_user(username, password)
+                if self.dbi.get_uid(username) != False:
+                    uid = self.dbi.get_uid(username)
+                    self.dbi.insert_worker(uid, username, password)
+                    self.cache.set(username, password)
+                    return True
        
        log.info("Authentication for %s failed" % username)
        return False