diff options
-rw-r--r-- | cupinder.py | 2 | ||||
-rw-r--r-- | db.py | 23 |
2 files changed, 22 insertions, 3 deletions
diff --git a/cupinder.py b/cupinder.py index e91fc21..2db4237 100644 --- a/cupinder.py +++ b/cupinder.py @@ -92,7 +92,7 @@ if __name__=='__main__': lid = LanguageIdentifier() - profiles = db.load_new_profiles() + profiles = db.load_unknown_profiles() remaining = tinder.get_remaining_likes() @@ -207,7 +207,7 @@ class HumanKind(object): return new - def load_new_profiles(self): + def load_unknown_profiles(self): """Load all new Tinder profiles.""" new = [] @@ -215,7 +215,26 @@ class HumanKind(object): values = (False, False, False) cursor = self._db.cursor() - cursor.execute('SELECT uid FROM People WHERE liked = ? AND superliked = ? AND passed = ?', values) + cursor.execute('SELECT uid FROM People WHERE liked = ? AND passed = ? AND superliked = ?', values) + + for found in cursor.fetchall(): + + profile = self.load_profile(found['uid']) + + new.append(profile) + + return new + + + def load_new_profiles(self): + """Load all new Tinder profiles.""" + + new = [] + + values = (False, ) + + cursor = self._db.cursor() + cursor.execute('SELECT uid FROM People WHERE extracted = ?', values) for found in cursor.fetchall(): |