Make fake_package more resilient

gratipay · Jun 9, 2017 · 6a8b4cb · 6a8b4cb
1 parent e22cfe7
commit 6a8b4cb
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/gratipay/fake_data.py b/gratipay/fake_data.py
@@ -246,13 +246,16 @@ def random_country_id(db):
 
 
 def fake_package(db):
-    insert_fake_data( db
-                    , 'packages'
-                    , package_manager='npm'
-                    , name=faker.word()
-                    , description=fake_sentence(stop=50)
-                    , emails=[faker.email() for i in range(random.choice(range(10)))]
-                     )
+    try:
+        insert_fake_data( db
+                        , 'packages'
+                        , package_manager='npm'
+                        , name=faker.word()
+                        , description=fake_sentence(stop=50)
+                        , emails=[faker.email() for i in range(random.choice(range(10)))]
+                         )
+    except IntegrityError:
+        return fake_package(db)
 
 
 def prep_db(db):