fix db migration

5ca6a6b3 · Romain Courteaud · 25f6d7d5 · 5ca6a6b3
Commit 5ca6a6b3 authored Jan 09, 2020 by Romain Courteaud
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

src/surykatka/db.py src/surykatka/db.py +6 -2

No files found.
--- a/src/surykatka/db.py
+++ b/src/surykatka/db.py
@@ -115,7 +115,7 @@ class LogDB:
            ip = peewee.TextField()
            url = peewee.TextField()
            status_code = peewee.IntegerField()
-            total_seconds = peewee.FloatField()
+            total_seconds = peewee.FloatField(default=0)

            class Meta:
                primary_key = peewee.CompositeKey("status", "ip", "url")
@@ -156,7 +156,11 @@ class LogDB:
                    # version 2 without the http total_seconds column
                    migrator = SqliteMigrator(self._db)
                    migrate(
-                        migrator.add_column("HttpCodeChange", "total_seconds")
+                        migrator.add_column(
+                            "HttpCodeChange",
+                            "total_seconds",
+                            self.HttpCodeChange.total_seconds,
+                        )
                    )

                if db_version >= expected_version: