znote 1.2.8__tar.gz → 1.4.0__tar.gz

{znote-1.2.8/znote.egg-info → znote-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: znote
-Version: 1.2.8
+Version: 1.4.0
 Summary: Vault de prise de note
 Author: 
 License: mit
@@ -168,6 +168,7 @@ La fonctionnalité de "channel" permet de synchroniser des fichiers entre plus
 | `channel <nom> --tree`         | Affiche l'arborescence et la liste des fichiers contenus dans le channel.                                                              | `znote channel mon-canal --tree`                                       |
 | `publish <channel> <fichier>`  | Publie un fichier sur un channel. Si le fichier existe déjà, il est mis à jour.                                                        | `znote publish mon-canal ./rapport.pdf`                                |
 | `publish <channel> --purge`    | Supprime les fichiers du channel qui ne sont pas présents dans la publication actuelle.                                                | `znote publish mon-canal --purge`                                      |
+| `unpublish <channel>`  | Supprime un channel.                                                         | `znote unpublish mon-canal`                                |
 | `fetch <channel> (<path>)`              | Récupère (télécharge) tous les fichiers d'un channel dans le répertoire local.                                                         | `znote fetch mon-canal`                                                |
 | `fetch <channel> --force`      | Force le téléchargement et l'écrasement des fichiers locaux même s'ils existent déjà.                                                    | `znote fetch mon-canal --force`                                        |
 | `fetch <channel> --purge`      | Supprime les fichiers locaux qui ne sont pas présent dans le channel.                                                    | `znote fetch mon-canal --purge`                                        |

{znote-1.2.8 → znote-1.4.0}/README.md

@@ -138,6 +138,7 @@ La fonctionnalité de "channel" permet de synchroniser des fichiers entre plusie
 | `channel <nom> --tree`         | Affiche l'arborescence et la liste des fichiers contenus dans le channel.                                                              | `znote channel mon-canal --tree`                                       |
 | `publish <channel> <fichier>`  | Publie un fichier sur un channel. Si le fichier existe déjà, il est mis à jour.                                                        | `znote publish mon-canal ./rapport.pdf`                                |
 | `publish <channel> --purge`    | Supprime les fichiers du channel qui ne sont pas présents dans la publication actuelle.                                                | `znote publish mon-canal --purge`                                      |
+| `unpublish <channel>`  | Supprime un channel.                                                         | `znote unpublish mon-canal`                                |
 | `fetch <channel> (<path>)`              | Récupère (télécharge) tous les fichiers d'un channel dans le répertoire local.                                                         | `znote fetch mon-canal`                                                |
 | `fetch <channel> --force`      | Force le téléchargement et l'écrasement des fichiers locaux même s'ils existent déjà.                                                    | `znote fetch mon-canal --force`                                        |
 | `fetch <channel> --purge`      | Supprime les fichiers locaux qui ne sont pas présent dans le channel.                                                    | `znote fetch mon-canal --purge`                                        |

{znote-1.2.8 → znote-1.4.0}/znote/client.py

@@ -44,23 +44,38 @@ class Client():
 
     #Réception des données
     @impmagic.loader(
-        {'module': 'select'}
+        {'module': 'select'},
+        {'module': 'time'}
     )
     def receive(self):
         data = b""
-        while True:
-            #Pour éviter les exceptions avec le socket non bloquant
-            #ready_to_read, _, _ = select.select([self.connexion], [], [], 1)
-            #if ready_to_read:
-            chunk = self.connexion.recv(2048)
 
-            if chunk and len(chunk)<2048:
-                return data+chunk
+        #Boucle pour récupérer l'ensemble des chunk d'un message
+        while True:
+            #Pour éviter les exceptions avec le socket non 
+            #A voir si ça crash pas si on reçoit de gros fichiers
+            ready_to_read, _, _ = select.select([self.connexion], [], [], 1)
+            if ready_to_read:
+                chunk = self.connexion.recv(2048)
+
+                if len(chunk):
+                    if not chunk.endswith(b'EOF'):
+                        data += chunk
+                    else:
+                        return data+chunk[:-3]
+                else:
+                    time.sleep(0.2)
+                """
+                if chunk:
+                    print(chunk)
+                if chunk and len(chunk)<2048:
+                    return data+chunk
 
-            if not chunk:
-                return data
+                if not chunk:
+                    return data
 
-            data += chunk
+                data += chunk
+                """
 
 
     @impmagic.loader(
@@ -93,7 +108,7 @@ class Client():
         # Sauvegarde du IV, de la clé symétrique chiffrée et du fichier chiffré
         message_cipher = iv + sym_key_encrypted + ciphertext
 
-        self.connexion.send(message_cipher)
+        self.connexion.send(message_cipher + b'EOF')
 
 
     #Premier appel avec le serveur
@@ -112,7 +127,7 @@ class Client():
             payload = {'method': 'sendKey', 'content': __main__.settings.PubKey.decode()}
             logs("Envoi de la clé public au serveur", component="client")
 
-        self.connexion.send(encrypt(json.dumps(payload).encode()))
+        self.connexion.send(encrypt(json.dumps(payload).encode()) + b'EOF')
 
         try:
             message_recu = self.receive()
@@ -225,7 +240,6 @@ class Client():
             self.well_come = True
 
         message = None
-
         #if not __main__.settings.pubserv_set:
         #if not __main__.settings.serverhash:
 
@@ -480,6 +494,15 @@ class Client():
                     if len(parameter)>2:
                         command['path'] = parameter[2]
 
+            elif instruction=="unpublish":
+                if len(parameter)>1:
+                    command = {
+                        'method': 'channel',
+                        'action': 'unpublish',
+                        'channel_name': parameter[1],
+                        'token': __main__.settings.token
+                    }
+
             elif instruction=="publish":
                 channel_name = parameter[1]
                 path = parameter[2]
@@ -662,7 +685,7 @@ class Client():
                         file_path = message['path']
                     else:
                         file_path = "."
-                    print(file_path)
+
                     diff_output = build_diff_tree(file_path, message["tree"])
                     print_tree(diff_output)
             

{znote-1.2.8 → znote-1.4.0}/znote/server.py

@@ -120,7 +120,7 @@ class ThreadClient(threading.Thread):
         message_cipher = iv + sym_key_encrypted + ciphertext
 
         try:
-            self.connexion.send(message_cipher)
+            self.connexion.send(message_cipher + b'EOF')
         except:
             logs(f"[Thread:{self.name}] - Echec de l'envoi", "error")
             self.kill_thread()
@@ -128,7 +128,8 @@ class ThreadClient(threading.Thread):
 
     #Réception des données
     @impmagic.loader(
-        {'module': 'select'}
+        {'module': 'select'},
+        {'module': 'time'}
     )
     def receive(self):
         data = b""
@@ -141,6 +142,16 @@ class ThreadClient(threading.Thread):
             if ready_to_read:
                 chunk = self.connexion.recv(2048)
 
+                if len(chunk):
+                    if not chunk.endswith(b'EOF'):
+                        data += chunk
+                    else:
+                        return data+chunk[:-3]
+                else:
+                    time.sleep(0.2)
+                """
+                if chunk:
+                    print(chunk)
                 if chunk and len(chunk)<2048:
                     return data+chunk
 
@@ -148,6 +159,7 @@ class ThreadClient(threading.Thread):
                     return data
 
                 data += chunk
+                """
 
 
     #Déchiffrement du message du client
@@ -230,13 +242,13 @@ class ThreadClient(threading.Thread):
 
                                     logs(f"[{self.get_name_user()}] - Clé public reçu")
                                     payload = {'method': 'sendKey', 'content': __main__.settings.PubKey.decode()}
-                                    self.connexion.send(self.encrypt(json.dumps(payload).encode()))
+                                    self.connexion.send(self.encrypt(json.dumps(payload).encode()) + b'EOF')
                                     logs(f"[{self.get_name_user()}] - Connecté")
 
                                 else:
                                     logs(f'[{self.get_name_user()}] - Clé public invalide', 'error')
                                     self.key=None
-                                    self.connexion.send("00000")
+                                    self.connexion.send("00000" + b'EOF')
                             
                             elif msgClient['method']=="welcome":    #Récupération de la clé publique du client grâce au token
                                 logs(f"[{self.get_name_user()}] - Welcome")
@@ -266,7 +278,7 @@ class ThreadClient(threading.Thread):
                                     payload = {'method': 'welcome', 'message': 'Bad request', 'status': 0}
                                     logs(f"[{self.get_name_user()}] - Bad request", "error")
 
-                                self.connexion.send(self.encrypt(json.dumps(payload).encode()))
+                                self.connexion.send(self.encrypt(json.dumps(payload).encode()) + b'EOF')
 
 
                         elif self.key!=None:
@@ -709,7 +721,33 @@ class ThreadClient(threading.Thread):
                                                             payload = {'method': 'channel', 'action': 'tree', 'list_files': list_files}
 
                                                         else:
-                                                            payload = {'method': 'notify', 'message': f"channel {message['channel_name']} introuvable", 'status_code': 'error'}  
+                                                            payload = {'method': 'notify', 'message': f"channel {message['channel_name']} introuvable", 'status_code': 'error'}
+
+                                                elif message['action'] == "unpublish":
+                                                    if "channel_name" in message:
+                                                        channel = __main__.settings.orm.Channel.get(name=message['channel_name'])
+                                                        if channel:
+                                                            channel_id = channel.id
+
+                                                            # Supprimer les publications liées
+                                                            query = __main__.settings.orm.PubChannel.select(lambda p: p.channel_id == channel_id)
+                                                            for pub in query:
+                                                                pub.delete()
+
+                                                            # Supprimer le channel
+                                                            channel.delete()
+
+                                                            pony.commit()
+
+                                                            logs(f"[{self.get_name_user()}] - channel: {message['channel_name']} supprimé avec ses publications", "success")
+                                                            payload = {'method': 'notify', 'message': f"channel {message['channel_name']} supprimé", 'status_code': 'success'}
+                                                        else:
+                                                            logs(f"[{self.get_name_user()}] - channel: {message['channel_name']} introuvable pour suppression", "error")
+                                                            payload = {'method': 'notify', 'message': f"channel {message['channel_name']} introuvable", 'status_code': 'error'}
+                                                    else:
+                                                        logs(f"[{self.get_name_user()}] - 'channel_name' manquant pour unpublish", "error")
+                                                        payload = {'method': 'notify', 'message': "'channel_name' manquant pour unpublish", 'status_code': 'error'}
+
 
 
                                     else:

{znote-1.2.8 → znote-1.4.0/znote.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: znote
-Version: 1.2.8
+Version: 1.4.0
 Summary: Vault de prise de note
 Author: 
 License: mit
@@ -168,6 +168,7 @@ La fonctionnalité de "channel" permet de synchroniser des fichiers entre plus
 | `channel <nom> --tree`         | Affiche l'arborescence et la liste des fichiers contenus dans le channel.                                                              | `znote channel mon-canal --tree`                                       |
 | `publish <channel> <fichier>`  | Publie un fichier sur un channel. Si le fichier existe déjà, il est mis à jour.                                                        | `znote publish mon-canal ./rapport.pdf`                                |
 | `publish <channel> --purge`    | Supprime les fichiers du channel qui ne sont pas présents dans la publication actuelle.                                                | `znote publish mon-canal --purge`                                      |
+| `unpublish <channel>`  | Supprime un channel.                                                         | `znote unpublish mon-canal`                                |
 | `fetch <channel> (<path>)`              | Récupère (télécharge) tous les fichiers d'un channel dans le répertoire local.                                                         | `znote fetch mon-canal`                                                |
 | `fetch <channel> --force`      | Force le téléchargement et l'écrasement des fichiers locaux même s'ils existent déjà.                                                    | `znote fetch mon-canal --force`                                        |
 | `fetch <channel> --purge`      | Supprime les fichiers locaux qui ne sont pas présent dans le channel.                                                    | `znote fetch mon-canal --purge`                                        |