From 8f74fb006c6b754efcca0328b87bd72aea51bd5c Mon Sep 17 00:00:00 2001
From: IDONTSUDO <idontsudo@gmail.com>
Date: Mon, 4 Sep 2023 23:07:34 +0300
Subject: [PATCH] adding new logic to test script

---
 .../python-modules/test-script/default.nix    | 44 ++++++++++++-------
 1 file changed, 28 insertions(+), 16 deletions(-)

diff --git a/pkgs/development/python-modules/test-script/default.nix b/pkgs/development/python-modules/test-script/default.nix
index 837bdb6..18a1d35 100644
--- a/pkgs/development/python-modules/test-script/default.nix
+++ b/pkgs/development/python-modules/test-script/default.nix
@@ -2,24 +2,36 @@
 writers.writePython3Bin "test-script" { libraries = []; } ''
 import sys
 import os
+import json
 
 
-def my_function():
+def main():
     arg = sys.argv[1]
-    number = int(arg)
-    if (number == 1):
-        os.makedirs('13')
-    if (number == 2):
-        raise Exception("error")
-    if (number == 3):
-        print(200)
-        print(200)
-        print(200)
-        print(200)
-        print(200)
-        print(200)
+
+    data = json.loads(arg)
+    paths = data.get('filesMeta')
+    path = data.get('path')
+    for el in paths:
+        if (el.get('path') is None and el.get('type') == 'folder'):
+            if (not os.path.exists((path + el.get('name')))):
+                os.makedirs(path + el.get('name'))
+        else:
+            filePath = el.get('path')
+            type = el.get('type')
+            name = el.get('name')
+            rewrite = el.get('rewrite')
+            newPath = path + filePath + '/'
+            if (type == 'file'):
+                if (rewrite and not os.path.exists((newPath + name))):
+                    file = open(newPath + name, 'w')
+                    file.write(str(el))
+                    file.close()
+            if (type == 'folder'):
+                if (rewrite and not os.path.exists((newPath + name))):
+                    os.makedirs(newPath + name)
 
 
-if __name__ == '__main__':
-    my_function()
-''
\ No newline at end of file
+main()
+''
+
+ 
\ No newline at end of file