Skip to content

Commit 83aa91d

Browse files
committed
sql: Create table if it is not present
1 parent d3515f8 commit 83aa91d

File tree

1 file changed

+3
-0
lines changed
  • src/flask_track_usage/storage

1 file changed

+3
-0
lines changed

src/flask_track_usage/storage/sql.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ def set_up(self, engine=None, metadata=None, table_name="flask_usage",
7474
.. versionchanged:: 1.1.0
7575
xforwardfor column added directly after remote_addr
7676
.. versionchanged:: 2.0.0
77+
table is created if it does not already exist
7778
added summary tables
7879
"""
7980

@@ -112,6 +113,8 @@ def set_up(self, engine=None, metadata=None, table_name="flask_usage",
112113
sql.Column('username', sql.String(128)),
113114
sql.Column('track_var', sql.String(128))
114115
)
116+
# Create the table if it does not exist
117+
self.track_table.create(bind=self._eng)
115118
else:
116119
self._metadata.reflect(bind=self._eng)
117120
self.track_table = self._metadata.tables[table_name]

0 commit comments

Comments
 (0)