diff options
Diffstat (limited to 'fetch.py')
-rwxr-xr-x | fetch.py | 13 |
1 files changed, 8 insertions, 5 deletions
@@ -13,16 +13,19 @@ from pyaggr3g470r.models import User if __name__ == "__main__": # Point of entry in execution mode + users, feed_id = [], None try: - feed_id = int(sys.argv[2]) + users = User.query.filter(User.id == int(sys.argv[1])).all() except: - feed_id = None + users = User.query.all() + finally: + if users == []: + users = User.query.all() - users = [] try: - users = User.query.filter(User.email == sys.argv[1]).all() + feed_id = int(sys.argv[2]) except: - users = User.query.all() + feed_id = None for user in users: if user.activation_key == "": |