Skip to content

Commit f935fb7

Browse files
committed
Update reset debug route and added images for database population
1 parent 9cf77c4 commit f935fb7

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

backend/PyMatcha/routes/api/debug.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from flask_jwt_extended import current_user
88
from flask_jwt_extended import jwt_required
99
from PyMatcha import redis
10+
from PyMatcha.models.image import Image
1011
from PyMatcha.models.like import Like
1112
from PyMatcha.models.match import Match
1213
from PyMatcha.models.message import Message
@@ -141,6 +142,7 @@ def delete_tables():
141142
User.drop_table()
142143
Tag.drop_table()
143144
Message.drop_table()
145+
Image.drop_table()
144146

145147
Match.create_table()
146148
Like.create_table()
@@ -149,6 +151,7 @@ def delete_tables():
149151
User.create_table()
150152
Tag.create_table()
151153
Message.create_table()
154+
Image.create_table()
152155
return "", 204
153156

154157

backend/PyMatcha/utils/populate_database.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import Geohash
77
import lorem
8+
from PyMatcha.models.image import Image
89
from PyMatcha.models.tag import Tag
910
from PyMatcha.models.user import get_user
1011
from PyMatcha.models.user import User
@@ -73,6 +74,7 @@ def populate_users(amount=150, drop_user_table=False):
7374

7475
for tag in tags:
7576
Tag.create(name=tag, user_id=u.id)
77+
Image.create(user_id=u.id, link=user.get_picture(), is_primary=True)
7678
except ConflictError:
7779
pass # Pass on the conflict error, this user wont be created because the username is taken. Who cares ?
7880

0 commit comments

Comments
 (0)