sciveo 0.1.15__tar.gz → 0.1.16__tar.gz

{sciveo-0.1.15 → sciveo-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.15
+Version: 0.1.16
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

sciveo-0.1.16/sciveo/common/tools/compress.py

@@ -0,0 +1,26 @@
+import json
+import gzip
+import base64
+
+from sciveo.common.tools.logger import *
+
+
+class CompressJsonData:
+  def __init__(self) -> None:
+    pass
+
+  def compress(self, data):
+    json_data = json.dumps(data).encode('utf-8')
+    compressed_data = gzip.compress(json_data)
+    encoded_data = base64.b64encode(compressed_data).decode('utf-8')
+    debug(type(self).__name__, "compress", len(json_data), "->", len(encoded_data))
+    return encoded_data
+
+  def decompress(self, encoded_data):
+    compressed_data = base64.b64decode(encoded_data)
+    decompressed_data = gzip.decompress(compressed_data)
+    json_data = decompressed_data.decode('utf-8')
+    data = json.loads(json_data)
+    debug(type(self).__name__, "decompress", len(encoded_data), "->", len(json_data))
+    return data
+

sciveo-0.1.16/sciveo/common/tools/crypto.py

@@ -0,0 +1,57 @@
+import json
+import base64
+from Crypto.Cipher import AES
+from Crypto.Random import get_random_bytes
+
+from sciveo.common.tools.logger import *
+
+
+class CryptoJsonData:
+  def __init__(self, base64_key=None) -> None:
+    if base64_key is None:
+      self.gen_key()
+    else:
+      self.key = base64.b64decode(base64_key)
+
+  def get_key(self):
+    return base64.b64encode(self.key).decode("utf-8")
+
+  def gen_key(self):
+    self.key = get_random_bytes(32)
+    return self.get_key()
+
+  def read_key(self, key_path):
+    with open(key_path, 'rb') as fp:
+      self.key = fp.read(32)
+
+  def write_key(self, key_path):
+    with open(key_path, 'wb') as fp:
+      fp.write(self.key)
+
+  def encrypt(self, plaintext, mode=AES.MODE_GCM):
+    iv = get_random_bytes(12)
+    cipher = AES.new(self.key, mode, nonce=iv)
+    data_encrypted, auth_tag = cipher.encrypt_and_digest(plaintext.encode())
+    return {
+      "encrypted": base64.b64encode(data_encrypted).decode("utf-8"),
+      "auth_tag": base64.b64encode(auth_tag).decode("utf-8"),
+      "iv": base64.b64encode(iv).decode("utf-8")
+    }
+
+  def decrypt(self, cipher_data, mode=AES.MODE_GCM):
+    cipher_text = base64.b64decode(cipher_data["encrypted"])
+    auth_tag = base64.b64decode(cipher_data["auth_tag"])
+    iv = base64.b64decode(cipher_data["iv"])
+    cipher = AES.new(self.key,  mode, nonce=iv)
+    return cipher.decrypt_and_verify(cipher_text, auth_tag).decode("utf-8")
+
+  def encrypt_json(self, data):
+    data = json.dumps(data)
+    data = base64.b64encode(data.encode()).decode("utf-8")
+    return self.encrypt(data)
+
+  def decrypt_json(self, encrypted_data):
+    data = self.decrypt(encrypted_data)
+    data = base64.b64decode(data)
+    data = json.loads(data)
+    return data

{sciveo-0.1.15 → sciveo-0.1.16}/sciveo/common/tools/daemon.py

@@ -10,6 +10,7 @@
 #
 
 import threading
+import signal
 import time
 
 from sciveo.common.tools.logger import *
@@ -17,6 +18,22 @@ from sciveo.common.tools.synchronized import ListQueue
 from sciveo.api.upload import APIFileUploader
 
 
+class SignalHandler:
+  def __init__(self, daemons):
+    self.daemons = daemons
+
+    signal.signal(signal.SIGINT, self.handler_int)
+    signal.signal(signal.SIGUSR1, self.handler_usr1)
+
+  def handler_int(self, signum, frame):
+    info(type(self).__name__, "INT signal received", signum, frame)
+    for daemon in self.daemons:
+      daemon.stop()
+
+  def handler_usr1(self, signum, frame):
+    info(type(self).__name__, "USR1 signal received", signum, frame)
+
+
 class DaemonBase:
   def __init__(self, num_threads=1, period=0):
     self.is_running = False

sciveo-0.1.16/sciveo/common/tools/random.py

@@ -0,0 +1,22 @@
+import random
+import string
+
+
+def random_password(length):
+  lower = string.ascii_lowercase
+  upper = string.ascii_uppercase
+  digits = string.digits
+  special = '!@#$%&'
+
+  # Ensure the password contains at least one character from each set
+  password = [
+    random.choice(lower),
+    random.choice(upper),
+    random.choice(digits),
+    random.choice(special)
+  ]
+
+  all_chars = lower + upper + (2 * digits) + (3 * special)
+  password += random.choices(all_chars, k=length - 4)
+  random.shuffle(password)
+  return ''.join(password)

sciveo-0.1.16/sciveo/version.py

@@ -0,0 +1,2 @@
+
+__version__ = '0.1.16'

{sciveo-0.1.15 → sciveo-0.1.16}/sciveo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.15
+Version: 0.1.16
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.15 → sciveo-0.1.16}/sciveo.egg-info/SOURCES.txt

@@ -18,11 +18,14 @@ sciveo/common/model.py
 sciveo/common/optimizers.py
 sciveo/common/sampling.py
 sciveo/common/tools/__init__.py
+sciveo/common/tools/compress.py
 sciveo/common/tools/configuration.py
+sciveo/common/tools/crypto.py
 sciveo/common/tools/daemon.py
 sciveo/common/tools/formating.py
 sciveo/common/tools/hardware.py
 sciveo/common/tools/logger.py
+sciveo/common/tools/random.py
 sciveo/common/tools/synchronized.py
 sciveo/common/tools/timers.py
 sciveo/content/__init__.py
@@ -34,7 +37,9 @@ sciveo/monitoring/__init__.py
 sciveo/monitoring/monitor.py
 sciveo/monitoring/network.py
 sciveo/monitoring/start.py
+test/test_compress.py
 test/test_configuration.py
+test/test_crypto.py
 test/test_monitoring.py
 test/test_runner.py
 test/test_sampling.py

{sciveo-0.1.15 → sciveo-0.1.16}/sciveo.egg-info/requires.txt

@@ -4,6 +4,7 @@ requests>=0.0.0
 [all]
 psutil>=0.0.0
 netifaces>=0.0.0
+pycryptodome>=0.0.0
 
 [mon]
 psutil>=0.0.0

{sciveo-0.1.15 → sciveo-0.1.16}/setup.py

@@ -32,6 +32,7 @@ setup(
       'all': [
         'psutil>=0.0.0',
         'netifaces>=0.0.0',
+        'pycryptodome>=0.0.0'
       ]
     },
     py_modules=['sciveo'],

sciveo-0.1.16/test/test_compress.py

@@ -0,0 +1,16 @@
+import unittest
+import json
+
+from sciveo.common.tools.compress import *
+
+
+class TestEncoders(unittest.TestCase):
+  def test_compress_json(self):
+    zip = CompressJsonData()
+    j1 = {"AAA": [1, 2, 3], "BBB": {"CCC": 1, "DDD": 2, "EEE": 3}, "CC": 3}
+    j2 = zip.decompress(zip.compress(j1))
+    self.assertEqual(json.dumps(j1, sort_keys=True), json.dumps(j2, sort_keys=True))
+
+
+if __name__ == '__main__':
+  unittest.main()

sciveo-0.1.16/test/test_crypto.py

@@ -0,0 +1,34 @@
+import unittest
+
+from sciveo.common.tools.crypto import *
+
+
+class TestCrypto(unittest.TestCase):
+  def test_string(self):
+    c = CryptoJsonData()
+    m1 = "test message string encrypt and decrypt"
+    m2 = c.decrypt(c.encrypt(m1))
+    self.assertEqual(m1, m2)
+
+  def test_json(self):
+    c = CryptoJsonData()
+    j1 = {"A": 1, "B": []}
+    j2 = c.decrypt_json(c.encrypt_json(j1))
+    self.assertEqual(json.dumps(j1, sort_keys=True), json.dumps(j2, sort_keys=True))
+
+  def test_big_json(self):
+    c = CryptoJsonData()
+    j1 = {"test": 1, "L": [1,2,3,4], "W": "A", "B": "C", "D1": {1: 2, 2: 3, 3: 4}, "L2": [[1,2,3], [3,4,5]]}
+    j2 = c.decrypt_json(c.encrypt_json(j1))
+    self.assertEqual(json.dumps(j1, sort_keys=True), json.dumps(j2, sort_keys=True))
+
+  def test_key(self):
+    key = "OZ1IIdAh3b6h+E7fgIGo33rRsNT8Vg428GVC4FAGZGM="
+    c = CryptoJsonData(key)
+    m1 = "test message string encrypt and decrypt"
+    m2 = c.decrypt(c.encrypt(m1))
+    self.assertEqual(m1, m2)
+
+
+if __name__ == '__main__':
+    unittest.main()

sciveo-0.1.15/sciveo/version.py

@@ -1,2 +0,0 @@
-
-__version__ = '0.1.15'