Added an incorrect password prompt.

cyberdb/data/datas.py

@@ -3,7 +3,6 @@
 '''
 
 
-import base64
 import pickle
 
 from obj_encrypt import Secret
@@ -57,20 +56,7 @@ def data_to_obj(self, data):
 
             try:
                 data = self._secret.decrypt(data)
-            except UnicodeDecodeError:
-                return {
-                    'code': 2,
-                    'errors-code': errors_code[2]
-                }
-
-            # Check if the dictionary is intact.
-            if type(data) != type(dict()):
-                return {
-                    'code': 2,
-                    'errors-code': errors_code[2]
-                }
-            elif type(data.get('content')) != type(b'') or not data.get('header') or not \
-                data.get('header').get('signature'):
+            except (UnicodeDecodeError, KeyError):
                 return {
                     'code': 2,
                     'errors-code': errors_code[2]
@@ -130,10 +116,8 @@ def obj_to_data(self, obj):
 
             data['header']['signature'] = self._signature.encrypt(data['content'])
             data = self._secret.encrypt(data)
-            # data = base64.b64encode(data)
 
         else:
-            # data = base64.b64encode(pickle.dumps(obj))
             data = pickle.dumps(obj)
 
         return data

cyberdb/network/__init__.py

@@ -102,7 +102,7 @@ def read(self) -> dict:
 
         # The client actively disconnects.
         if number_of_times == b'':
-            raise DisconCyberDBError('The TCP connection was disconnected by the other end.')
+            raise DisconCyberDBError('The TCP connection was disconnected by the other end. It is also possible that the password entered by the client is incorrect.')
 
         r = self._dp.data_to_obj(number_of_times)
         if r['code'] != 1:

setup.py

@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="CyberDB",
-    version="0.7.0",
+    version="0.7.1",
     author="Cyberbolt",
     author_email="[email protected]",
     description="CyberDB is a lightweight Python in-memory database. It is designed to use Python's built-in data structures Dictionaries, Lists for data storage, efficient communication through Socket TCP, and provide data persistence. This module can be used in hard disk database caching, Gunicorn inter-process communication, distributed computing and other fields.",