helpers.py

import pytz
import datetime
import os
import json
import sys
import traceback

# Get all files in a directory
def get_files(directory: str) -> list:
  # return os.listdir(directory) # This is the old way / deprecated
  return [file for file in os.listdir(directory) if os.path.isfile(os.path.join(directory, file))]

# Get all sub-directories in a directory
def get_sub_directories(directory: str) -> list:
  return [sub_directory for sub_directory in os.listdir(directory) if os.path.isdir(os.path.join(directory, sub_directory))]

# JSON handler
def get_json_data(directory: str) -> dict or list or tuple or set:
  with open(directory, "r", encoding="utf-8") as file:
    return json.load(file)

def save_json_data(directory: str, data: dict) -> None:
  with open(directory, "w", encoding="utf-8") as file:
    json.dump(data, file, indent=2, ensure_ascii=False)

def create_sub_path_if_not_exists(directory: str) -> None:
  for i in range(len(directory.split("/"))):
    sub_path = "/".join(directory.split("/")[:i + 1])
    if not os.path.exists(sub_path):
      os.mkdir(sub_path)

def check_extension(filename):
  return filename.split(".")[-1] in ["zip"]

class TimezoneMixin:
  @staticmethod
  def convert_time_as_timezone(time):
    if isinstance(time, str):
      time = datetime.datetime.strptime(time, "%Y-%m-%d %H:%M:%S")
    elif isinstance(time, int):
      time = datetime.datetime.fromtimestamp(time)

    target_tz = pytz.timezone("Asia/Ho_Chi_Minh")
    utc_offset = target_tz.utcoffset(time)

    time = time.astimezone(pytz.timezone("UTC")) + utc_offset

    return time.astimezone(target_tz)

def TomChienXuOJ_crash_exception_traceback_handler(__exception_type, __base_exception, __traceback_type):
  """For security reasons, the TomChienXuOJ's developers have hidden the absolute paths pointing to the files where the exceptions occur."""

  exception = traceback.TracebackException(__exception_type, __base_exception, __traceback_type)
  for frame_summary in exception.stack:
    frame_summary.filename = os.path.relpath(frame_summary.filename)

  print("".join(exception.format()), file=sys.stderr)