Skip to content

Commit 434fe13

Browse files
committed
Added fix for pop db in case of 503
1 parent 3c4837b commit 434fe13

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

backend/PyMatcha/utils/populate_database.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
import json
33
import os
44
import random
5+
from time import sleep
6+
from urllib.error import HTTPError
57

68
import Geohash
79
import lorem
@@ -25,7 +27,11 @@ def populate_users(amount=150, drop_user_table=False):
2527
if drop_user_table:
2628
User.drop_table()
2729
User.create_table()
28-
users = RandomUser.generate_users(amount=amount)
30+
try:
31+
users = RandomUser.generate_users(amount=amount)
32+
except HTTPError:
33+
sleep(10)
34+
users = RandomUser.generate_users(amount=amount)
2935
for user in users:
3036
gender = random.choice(["male", "female", "other"])
3137
if gender != "other":

0 commit comments

Comments
 (0)