Merge remote-tracking branch 'refs/remotes/origin/main'

refit/src/modules/librefit.py

@@ -0,0 +1,35 @@
+from .refit_logger import logger
+
+
+def get_standard_name_number(current_number, number_str_length):
+    """returns a number string filled to the length of the input number"""
+    current_number = str(current_number)
+    standard_name_number = str.zfill(current_number, number_str_length)
+    # logger.debug(f"standard_name_number={standard_name_number}")
+    return standard_name_number
+
+
+def get_standard_folder_name(name):
+    """returns the default name if name=None else, it returns the first
+    given name to the argument --name as astring"""
+    logger.debug("get_standard_folder_name() call")
+    if name is None:
+        logger.warning("No name assigned, continue with default.")
+        standard_folder_name = "directory"
+    else:
+        logger.debug(f"{name}, Type: {type(name)}, Length: {len(name)}")
+
+        # Checks if multiple names are passed to the --name argument
+        # if so, only the first one gets used.
+        if len(name) > 1:
+            _NAMES_WARNING = f"{len(name)} names given, only one required.\nContinuing with '{name[0]}' as name."
+            logger.warning(_NAMES_WARNING)
+            print(_NAMES_WARNING)
+            standard_folder_name = name[0]
+
+            logger.debug(f"{standard_folder_name}, Type: {type(standard_folder_name)}")
+        else:
+            standard_folder_name = name[0]
+            logger.debug("Continuing with expected input")
+
+    return standard_folder_name

refit/src/modules/refit_create.py

@@ -1,8 +1,7 @@
 import os
-import pathlib
-import sys
 
 from .refit_logger import logger
+from . import librefit
 
 logger.debug("Initiated refit_create.py")
 # ----------------------------------------------------------------------
@@ -42,38 +41,27 @@ class Refit_Create:
 
         logger.debug("in create_n_folders()")
 
-        # Sets the default directory name.
-        if name is None:
-            logger.debug("name argument is 'None', applying default name")
-            name = "directory"
-        else:
-            logger.debug(f"{name}, Type: {type(name)}, Length: {len(name)}")
-
-            # Checks if multiple names are passed to the --name argument
-            # if so, only the first one gets used.
-            if len(name) > 1:
-                _NAMES_WARNING = f"{len(name)} names given, only one required.\nContinuing with '{name[0]}' as name."
-                logger.warning(_NAMES_WARNING)
-                print(_NAMES_WARNING)
-            name = name[0]
-
-            logger.debug(f"{name}, Type: {type(name)}")
-
-        # Creating the length of the suffix.
+        # Creating the length of the suffix number_string.
         length_n = len(str(n))
+
+        # Get either the default folder name or the input name as string.
+        folder_name = librefit.get_standard_folder_name(name)
+
         logger.debug(f"Length of numbering string:{length_n}")
+        logger.debug(f"Folder name: {folder_name}")
 
         while n > 0:
-            # Creating the suffix string, filling the numbers to len of arg.n
+            # iterating down for the files number.
             i = str(n - 1)
-            number_string = str.zfill(i, length_n)
+
+            # Passing the number and the length of the string to get the string back.
+            number_string = librefit.get_standard_name_number(i, length_n)
 
             # Creating path for the folder
-            temp_name = f"{name}_{number_string}"
-            rfc_path = os.path.join(input, temp_name)
-
+            temp_name = f"{folder_name}_{number_string}"
+            folder_creation_path = os.path.join(input, temp_name)
             # Creating folder and subtracting n by one for the number_string
-            os.mkdir(rfc_path)
+            os.mkdir(folder_creation_path)
             n -= 1
 
     def rf_create_decider(self):

refit/src/refit.py

@@ -27,10 +27,7 @@ subparser = parser.add_subparsers(
 )
 
 # Create Parser Arguments
-create_parser = subparser.add_parser(
-    "create",
-    help="creates a new file/folder",
-)
+create_parser = subparser.add_parser("create", help="creates a new file/folder")
 create_parser.add_argument("-n", type=int, help="number of items")
 create_parser.add_argument("-i", "--input", help="input file")
 create_parser.add_argument(