From: gled <gled@remote-shell.net>
Date: Tue, 11 Jun 2019 17:48:49 +0000 (-0700)
Subject: implement force-remove in manage.py, should only be used when relay is not running
X-Git-Tag: 0.1.0~13^2
X-Git-Url: https://reisub.nsupdate.info/git/?a=commitdiff_plain;h=06d9d4c19e6ffe30dc049f9a9dba0c851a2057a3;hp=f34b42ae78ff03ed346e69086ec435cf420a35a9;p=relay.git%2F.git

implement force-remove in manage.py, should only be used when relay is not running
---

diff --git a/relay/manage.py b/relay/manage.py
index 542a008..4912fb1 100644
--- a/relay/manage.py
+++ b/relay/manage.py
@@ -1,6 +1,9 @@
 import asyncio
 import sys
+import simplejson as json
+
 from .actor import follow_remote_actor, unfollow_remote_actor
+from . import CONFIG
 from .database import DATABASE
 
 
@@ -34,11 +37,28 @@ def relay_unfollow():
 
     print('Sent unfollow message to:', target)
 
+def relay_forceremove():
+    if len(sys.argv) < 3:
+        print('usage: python3 -m relay.manage force-remove <target>')
+        exit()
+
+    target = sys.argv[2]
+
+    following = DATABASE.get('relay-list', [])
+
+    if target in following:
+        following.remove(target)
+        DATABASE['relay-list'] = following
+        with open('relay.jsonld', 'w') as f:
+            json.dump(DATABASE, f)
+        print('Removed target from DB:', target)
+
 
 TASKS = {
     'list': relay_list,
     'follow': relay_follow,
-    'unfollow': relay_unfollow
+    'unfollow': relay_unfollow,
+    'force-remove': relay_forceremove
 }