Auto reset on failure

README.md

@@ -30,9 +30,6 @@ Usage Example
 	http://localhost:62456/api/randbytes?length=8
 	< <20><><EFBFBD>E<EFBFBD><45>s_<73>b<EFBFBD><62><EFBFBD><EFBFBD>G<EFBFBD>
 
-	http://localhost:62456/api/reset
-	< OK
-
 
 ### WebSocket API
 	
@@ -45,8 +42,6 @@ Usage Example
 	< -1.6120135370
 	> RANDBYTES 16
 	< <20><><EFBFBD><EFBFBD><EFBFBD>9L).<2E>!:<3A><><EFBFBD>@Nh<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>d<EFBFBD><EFBFBD><EFBFBD><EFBFBD>_q<EFBFBD>
-	> RESET
-	< OK
 
 License
 -------

quanttp/__main__.py

@@ -57,12 +57,6 @@ def api_randbytes():
         return Response(str(e), status=400, content_type='text/plain')
 
 
-@app.route('/api/reset')
-def api_reset():
-    qng_wrapper.reset()
-    return Response('OK', content_type='text/plain')
-
-
 @sockets.route('/ws')
 def ws(websocket):
     while not websocket.closed:
@@ -77,9 +71,6 @@ def ws(websocket):
                 websocket.send(str(qng_wrapper.randnormal()))
             elif split_message[0] == 'randbytes':
                 websocket.send(qng_wrapper.randbytes(int(split_message[1])))
-            elif split_message[0] == 'reset':
-                qng_wrapper.reset()
-                websocket.send('OK')
         except ValueError as e:
             websocket.close(code=1003, message=str(e))
         except Exception as e:

quanttp/data/qng_wrapper.py

@@ -27,20 +27,33 @@ class QngWrapper:
         self.qng = win32com.client.Dispatch("QWQNG.QNG")
 
     def randint32(self):
-        self.qng.Clear()
-        return self.qng.RandInt32
+        try:
+            self.qng.Clear()
+            return self.qng.RandInt32
+        except:
+            self.qng.Reset()
+            return self.qng.RandInt32
 
     def randuniform(self):
-        self.qng.Clear()
-        return self.qng.RandUniform
+        try:
+            self.qng.Clear()
+            return self.qng.RandUniform
+        except:
+            self.qng.Reset()
+            return self.qng.RandUniform
 
     def randnormal(self):
-        self.qng.Clear()
-        return self.qng.RandNormal
+        try:
+            self.qng.Clear()
+            return self.qng.RandNormal
+        except:
+            self.qng.Reset()
+            return self.qng.RandNormal
 
     def randbytes(self, length):
-        self.qng.Clear()
-        return self.qng.RandBytes(length)
-
-    def reset(self):
-        self.qng.Reset()
+        try:
+            self.qng.Clear()
+            return self.qng.RandBytes(length)
+        except:
+            self.qng.Reset()
+            return self.qng.RandBytes(length)

quanttp/data/qng_wrapper_mock.py

@@ -35,6 +35,3 @@ class QngWrapperMock:
 
     def randbytes(self, length):
         return secrets.token_bytes(nbytes=length)
-
-    def reset(self):
-        pass