From 4e09ce75bccf0847730a291941020deefd9634f9 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 17 Jul 2024 21:32:53 +0100
Subject: [PATCH] Add restart on fail

---
 Makefile                     |  2 +-
 setup.py                     |  2 +-
 trustgraph/base/processor.py | 30 ++++++++++++++++--------------
 3 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/Makefile b/Makefile
index 3e3ba808..9dea4f59 100644
--- a/Makefile
+++ b/Makefile
@@ -1,6 +1,6 @@
 
 # VERSION=$(shell git describe | sed 's/^v//')
-VERSION=0.3.0
+VERSION=0.3.1
 
 all: container
 
diff --git a/setup.py b/setup.py
index d40ce093..c6c32f97 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ import os
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-version = "0.3.0"
+version = "0.3.1"
 
 setuptools.setup(
     name="trustgraph",
diff --git a/trustgraph/base/processor.py b/trustgraph/base/processor.py
index e214b320..4f035966 100644
--- a/trustgraph/base/processor.py
+++ b/trustgraph/base/processor.py
@@ -57,27 +57,29 @@ class BaseProcessor:
     @classmethod
     def start(cls, prog, doc):
 
-        parser = argparse.ArgumentParser(
-            prog=prog,
-            description=doc
-        )
+        while True:
 
-        cls.add_args(parser)
+            parser = argparse.ArgumentParser(
+                prog=prog,
+                description=doc
+            )
 
-        args = parser.parse_args()
-        args = vars(args)
+            cls.add_args(parser)
 
-        try:
+            args = parser.parse_args()
+            args = vars(args)
 
-            p = cls(**args)
-            p.run()
+            try:
 
-        except Exception as e:
+                p = cls(**args)
+                p.run()
 
-            print("Exception:", e, flush=True)
-            print("Will retry...", flush=True)
+            except Exception as e:
 
-            time.sleep(10)
+                print("Exception:", e, flush=True)
+                print("Will retry...", flush=True)
+
+                time.sleep(10)
 
 class Consumer(BaseProcessor):