From 52bea4e25fe84d4793d5b1df29dc356330c208d8 Mon Sep 17 00:00:00 2001
From: pcosson <pcosson@comitari.fr>
Date: Fri, 12 Jul 2024 07:12:38 +0000
Subject: [PATCH] Ajouter setup.py

---
 setup.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 setup.py

diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..9d03dae
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+from setuphelpers import *
+import sys
+
+def install():
+    pass
+
+
+def audit():
+
+    if sys.platform == "win32":  #windows
+        teamviewer_id = registry_readstring(HKEY_LOCAL_MACHINE,r'SOFTWARE/TeamViewer','ClientID')
+        if teamviewer_id:
+            WAPT.write_audit_data_if_changed("TeamViewer", "id", teamviewer_id)
+        else:
+            print("Error while retrieving TeamViewer id.")
+
+
+    elif sys.platform == "darwin":   #MacOS
+        import subprocess
+        cmd = ["/Applications/TeamViewer.app/Contents/MacOS/TeamViewer", "info"]
+        result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+
+        if result.returncode !=0:
+            print(f"Erreur lors de la requête: {result.stderr}")
+            #return 'Error'
+
+        for line in result.stdout.splitlines():
+            if "TeamViewer ID" in line:
+                teamviewer_id = line.split(":")[1].strip()
+                WAPT.write_audit_data_if_changed("TeamViewer", "id", teamviewer_id)
+
+    elif "linux" in sys.platform :      #Linux (values can be linux or linux2)
+        result = run("teamviewer info")
+        
+        for line in result.splitlines():
+            if "TeamViewer ID" in line:
+                teamviewer_id = line.split(":")[1].strip()
+                teamviewer_id = teamviewer_id.split(' ')
+                WAPT.write_audit_data_if_changed("TeamViewer", "id", teamviewer_id[2])
+    else:
+        print("Unsupported OS")
+        return 'ERROR'
+    return 'OK'
\ No newline at end of file