cmd app with progress

excel_code_generator.py

@@ -0,0 +1,114 @@
+import asyncio
+from enum import Enum
+import time
+import pandas as pd
+from typing import List
+import random
+from dataclasses import dataclass, field
+
+
+@dataclass
+class Numbers:
+    '''Data class for input and output numbers'''
+    input_numbers: List[int] = field(default_factory=list)
+    output_numbers: List[int] = field(default_factory=list)
+
+
+class ExcelCodeGenerator:
+    class Config:
+        def __init__(self, input_file_path: str, output_file_path: str, should_be_unique: bool, should_have_no_conflict_with_input_numbers: bool, digits: int):
+            self.execl_file_handler_config: ExcelFileHandler.Config = ExcelFileHandler.Config(
+                input_file_path, output_file_path)
+            self.unique_random_generator_config: UniqueRandomGenerator.Config = UniqueRandomGenerator.Config(
+                should_be_unique, should_have_no_conflict_with_input_numbers, digits)
+
+    def __init__(self, config: Config):
+        self.config: ExcelCodeGenerator.Config = config
+        self.numbers: Numbers = Numbers()
+        self.file_handler: ExcelFileHandler = ExcelFileHandler(
+            self.config.execl_file_handler_config, self.numbers)
+        self.generator: UniqueRandomGenerator = UniqueRandomGenerator(
+            self.config.unique_random_generator_config, self.numbers)
+
+    def do_the_work(self, show_progress: bool = True):
+        print("reading", end="\n", flush=True)
+        self.file_handler.read()
+        print("generating", end="", flush=True)
+        self.generator.generate()
+        print("\nwriting", end="\n", flush=True)
+        self.file_handler.write()
+
+
+class ExcelFileHandler:
+    class Config:
+        def __init__(self, input_file_path: str, output_file_path: str):
+            self.input_file_path: str = input_file_path
+            self.output_file_path: str = output_file_path
+            self._check()
+
+        def _check(self):
+            if self.input_file_path == self.output_file_path:
+                print("Error: input[{}] file is same as output[{}], choose different output name".format(
+                    self.input_file_path, self.output_file_path))
+                exit(-1)
+
+    def __init__(self, config: Config, numbers: Numbers):
+        self.config = config
+        self.numbers = numbers
+
+    def read(self):
+        xl = pd.ExcelFile(self.config.input_file_path)
+        sheet1 = xl.parse(0)
+
+        self.numbers.input_numbers.extend(
+            list(map(lambda x: int(x), sheet1.iloc[:, 0])))
+
+    def write(self):
+        df = pd.DataFrame({'input': self.numbers.input_numbers,
+                           'output': self.numbers.output_numbers})
+
+        writer = pd.ExcelWriter(self.config.output_file_path)
+        df.to_excel(writer, 'Sheet1', index=False)
+        writer.save()
+
+
+class UniqueRandomGenerator:
+    @dataclass
+    class Config:
+        should_be_unique: bool = True
+        should_have_no_conflict_with_input_numbers: bool = True
+        digits: int = 8
+
+    def __init__(self, config: Config, numbers: Numbers):
+        self.config = config
+        self.numbers = numbers
+
+    def _generate_n_digit_random(self):
+        lower_value: int = 10**(self.config.digits - 1)
+        if lower_value == 1:
+            lower_value = 0
+        upper_value: int = 10**self.config.digits - 1
+        number: int = random.randint(lower_value, upper_value)
+        return number
+
+    def generate(self):
+        number: int = 0
+        progress_prev: int = 0
+        progress_current: int = 0
+
+        while len(self.numbers.output_numbers) < len(self.numbers.input_numbers):
+            progress_current = int(len(self.numbers.output_numbers) // (len(self.numbers.input_numbers) // 100 ))
+            if progress_current > progress_prev:
+                if progress_current % 5 == 0:
+                    print(" %{} ".format(progress_current), end="", flush=True)
+                else:
+                    print(".", end="", flush=True)
+                progress_prev=progress_current
+            number=self._generate_n_digit_random()
+            if self.config.should_be_unique:
+                if number in self.numbers.output_numbers:
+                    continue
+            if self.config.should_have_no_conflict_with_input_numbers:
+                if number in self.numbers.input_numbers:
+                    continue
+            self.numbers.output_numbers.append(number)

main.py

@@ -0,0 +1,10 @@
+import excel_code_generator as ecg
+
+
+def excel_code_generator_runner():
+    config: ecg.ExcelCodeGenerator.Config = ecg.ExcelCodeGenerator.Config(
+        "input.xlsx", "output.xlsx", True, True, 8)
+    worker: ecg.ExcelCodeGenerator = ecg.ExcelCodeGenerator(config)
+    worker.do_the_work()
+
+excel_code_generator_runner()

tests.py

@@ -0,0 +1,35 @@
+import excel_code_generator as ecg
+
+
+def test_UniqueRandomGenerator():
+    urg_config: ecg.UniqueRandomGenerator.Config = ecg.UniqueRandomGenerator.Config(
+        True, True, 1)
+    numbers: ecg.Numbers = ecg.Numbers([1, 2, 3, 4])
+    urg: ecg.UniqueRandomGenerator = ecg.UniqueRandomGenerator(
+        urg_config, numbers)
+    urg.generate()
+    print(numbers.output_numbers)
+
+
+# test_UniqueRandomGenerator()
+
+def test_ExcelFileHandler():
+    config: ecg.ExcelFileHandler.Config = ecg.ExcelFileHandler.Config(
+        "input.xlsx", "output.xlsx")
+    numbers: ecg.Numbers = ecg.Numbers()
+    file_handler: ecg.ExcelFileHandler = ecg.ExcelFileHandler(config, numbers)
+    file_handler.read()
+    print(numbers.input_numbers)
+    print(numbers.output_numbers)
+
+
+# test_ExcelFileHandler()
+
+def test_ExcelCodeGenerator():
+    config: ecg.ExcelCodeGenerator.Config = ecg.ExcelCodeGenerator.Config(
+        "input.xlsx", "output.xlsx", True, True, 8)
+    worker: ecg.ExcelCodeGenerator = ecg.ExcelCodeGenerator(config)
+    worker.do_the_work()
+
+
+test_ExcelCodeGenerator()