From fbd9321fc50a7c73079c7f46e78d204dd6aadae1 Mon Sep 17 00:00:00 2001
From: Iain Learmonth <irl@sr2.uk>
Date: Fri, 22 Apr 2022 14:29:24 +0100
Subject: [PATCH] cli: import/export more tables

---
 app/cli/db.py               | 14 ++++++++------
 app/terraform/block_ooni.py |  2 +-
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/app/cli/db.py b/app/cli/db.py
index def809e..3d03437 100644
--- a/app/cli/db.py
+++ b/app/cli/db.py
@@ -6,17 +6,19 @@ import sys
 
 from app import app
 from app.extensions import db
-from app.models.base import Group
+from app.models.base import Group, MirrorList
 from app.models.bridges import BridgeConf
-from app.models.mirrors import Origin, Proxy
+from app.models.mirrors import Mirror, Origin, Proxy
 from app.models.alarms import Alarm
 
 models = {
-    "group": Group,
-    "origin": Origin,
-    "proxy": Proxy,
     "bridge": BridgeConf,
-    "alarm": Alarm
+    "alarm": Alarm,
+    "group": Group,
+    "list": MirrorList,
+    "mirror": Mirror,
+    "origin": Origin,
+    "proxy": Proxy
 }
 
 
diff --git a/app/terraform/block_ooni.py b/app/terraform/block_ooni.py
index 7beacea..e471e7b 100644
--- a/app/terraform/block_ooni.py
+++ b/app/terraform/block_ooni.py
@@ -8,7 +8,7 @@ import requests
 from app import app
 from app.extensions import db
 from app.models.alarms import Alarm, AlarmState
-from app.models.mirrors import Proxy
+from app.models.mirrors import Origin
 
 
 def check_origin(domain_name: str):