mirror of
https://github.com/ChronosX88/PyNesca.git
synced 2024-11-26 06:32:19 +00:00
24 lines
644 B
Python
24 lines
644 B
Python
|
from AbstractStorage import AbstractStorage
|
||
|
|
import json
|
||
|
|
from threading import RLock
|
||
|
|
|
||
|
|
|
||
|
|
class JSONStorage(AbstractStorage):
|
||
|
|
|
||
|
|
def __init__(self, path):
|
||
|
|
self.path = path
|
||
|
|
self.respdict = dict()
|
||
|
|
self.lock = RLock()
|
||
|
|
|
||
|
|
def put_responce(self, address, responce):
|
||
|
|
ip, port = address
|
||
|
|
if ip not in self.respdict.keys():
|
||
|
|
self.respdict[ip] = {"open": [], "close": []}
|
||
|
|
self.respdict[ip]["open" if responce != 0 else "close"].append(port)
|
||
|
|
|
||
|
|
def save(self):
|
||
|
|
print("saving")
|
||
|
|
with open(self.path, "w") as f:
|
||
|
|
json.dump(self.respdict, f)
|
||
|
|
self.respdict = {}
|