util.py

import json
import os

class Utils:
    def __init__(self,file_path):
        self.file_path = file_path

    def get_number_of_problems(self):
         with open(self.file_path, 'r') as file:
            data = json.load(file)
            return len(data)        
            
    def display_progress_bar(self, problem_number, problem_treated_num):
        if problem_number == 0:
            percentage = 100
        else:
            percentage = (problem_treated_num / problem_number) * 100
        
        progress_bar_length = 50

        num_dashes = int((percentage / 100) * progress_bar_length)
        
        progress_bar = "-" * num_dashes + " " * (progress_bar_length - num_dashes)
        
        print(f"\r[{progress_bar}] {percentage:.2f}%", end="")

    def ExtractProblemName(self, text):
        if not text:
            return None
        
        text_splitted = str(text).split(' ')
        
        if 'ago' in text_splitted:
            ago_index = text_splitted.index('ago')
            new_one = text_splitted[ago_index+1:]
            
            if 'Accepted' in new_one:
                accepted_index = new_one.index('Accepted')
                problem_name = ' '.join(new_one[:accepted_index])
                return problem_name
        
        return None
    
    def extract_language_name(self, text):
        if not text:
            return None
        
        text_splitted = text.split(' ')

        return text_splitted[-1]
        

    def remove_duplicates_from_json(self):
        with open(self.file_path, 'r') as file:
            data = json.load(file)

        if isinstance(data, list):
            seen_problems = set()
            unique_data = []

            for item in data:
                problem = item.get('problem')
                if problem not in seen_problems:
                    seen_problems.add(problem)
                    unique_data.append(item)
            
            data = unique_data
        else:
            print("Unsupported JSON format. The data must be a list of dictionaries.")
            return

        with open(self.file_path, 'w') as file:
            json.dump(data, file, indent=4)

    def get_language_suffix(self, language):
        language_suffixes = { # i generally code either with python or cpp or c in worst case lol but you can extended this dict as you wish
            "cpp": "cpp",
            "python": "py",
            "python3": "py",
            "c": "c",
            "java": "java",
            "javascript": "js",
            "typescript": "ts",
            "ruby": "rb",
            "swift": "swift",
            "golang": "go",
            "kotlin": "kt",
            "rust": "rs",
            "php": "php",
            "scala": "scala",
            "racket": "rkt",
            "erlang": "erl",
            "elixir": "ex",
            "haskell": "hs",
            "perl": "pl",
            "dart": "dart",
            "bash": "sh",
            "mysql": "sql",
            "mssql": "sql",
            "oraclesql": "sql",
            "csharp": "cs",
            "vb": "vb",
            "objectivec": "m",
            "groovy": "groovy"
        }

        return language_suffixes.get(language, "txt") # txt for default 

    def create_folders(self):
        with open(self.file_path, 'r') as file:
            data = json.load(file)

        base_path = "./problemset"

        if not os.path.exists(base_path):
            os.makedirs(base_path)

        for row in data:
            try:
                problem_name = row['problem'].replace(' ', '_')
                problem_path = os.path.join(base_path, problem_name)
                
                if not os.path.exists(problem_path):
                    os.makedirs(problem_path)
                
                language_suffix = self.get_language_suffix(row['language'])
                file_name = "solution." + language_suffix
                file_path = os.path.join(problem_path, file_name)
                with open(file_path, 'w') as file:
                    file.write(row['code'])
                
                print(f"Created folder and file for problem: {problem_name}")
            except Exception as e:
                print(f"Error: {e}")