Refactor folder creation logic to add leading zeros to AVP folder numbers
This commit is contained in:
parent
e19076efeb
commit
8f35f927c1
2 changed files with 1 additions and 71 deletions
211
GestionVersion2.3.py
Normal file
211
GestionVersion2.3.py
Normal file
|
|
@ -0,0 +1,211 @@
|
|||
# Todo:
|
||||
# - Make an AVP folder in the current directory whe creating a new project
|
||||
# - Make a BPE folder in the current directory whe creating a new project
|
||||
|
||||
|
||||
|
||||
import os
|
||||
import datetime
|
||||
|
||||
#If colorama is not installed, notify the user
|
||||
try:
|
||||
from colorama import init, Fore, Style
|
||||
except ImportError:
|
||||
print("Please install the colorama module using 'pip install colorama' to enable colored text.")
|
||||
Fore = Style = lambda x: x
|
||||
|
||||
debug_mode = True
|
||||
|
||||
# Initialize colorama
|
||||
init()
|
||||
|
||||
# Function to pause if debug mode is enabled
|
||||
def pause_debug(message="No message provided."):
|
||||
if debug_mode:
|
||||
input(Fore.YELLOW + "[DEBUG] " + message + " - Enter to continue.\n" + Style.RESET_ALL)
|
||||
|
||||
|
||||
|
||||
current_directory = os.path.dirname(os.path.abspath(__file__)) # Get the current directory
|
||||
os.chdir(current_directory)
|
||||
pause_debug(f"Current directory: {current_directory}")
|
||||
|
||||
avp_directory = os.path.join(current_directory, "AVP") # Set the source directory to the 'Projets' directory
|
||||
pause_debug(f"AVP directory: {avp_directory}")
|
||||
|
||||
def createNewProject():
|
||||
|
||||
|
||||
# Prompt the user for the project name
|
||||
project_name = input("Enter the project name:\n")
|
||||
pause_debug(f"Project name entered: {project_name}")
|
||||
|
||||
# Get the current year and month
|
||||
current_year = datetime.datetime.now().year
|
||||
current_month = datetime.datetime.now().strftime("%m")
|
||||
pause_debug(f"Current year and month: {current_year} {current_month}")
|
||||
|
||||
# Get all folders starting with the current year and month in the source directory
|
||||
pattern = f"{current_year} {current_month}"
|
||||
pause_debug(f"Pattern to search for: {pattern}")
|
||||
|
||||
folders = [f for f in os.listdir(current_directory) if os.path.isdir(os.path.join(current_directory, f)) and f.startswith(pattern)]
|
||||
if not folders:
|
||||
latest_number = 0
|
||||
else:
|
||||
latest_number = max(int(f.split()[2]) for f in folders)
|
||||
pause_debug(f"Folders matching pattern '{pattern}' found: {', '.join(folders)}")
|
||||
|
||||
# Find the folder with the largest number after the year and month
|
||||
if not folders:
|
||||
new_number = 1
|
||||
else:
|
||||
latest_number = max(int(f.split()[2]) for f in folders)
|
||||
new_number = latest_number + 1
|
||||
pause_debug(f"New folder number determined: {new_number}")
|
||||
|
||||
# Create the new folder name
|
||||
new_folder_name = f"{current_year} {current_month} {new_number} {project_name}"
|
||||
pause_debug(f"New folder name: {new_folder_name}")
|
||||
|
||||
# Create the new folder
|
||||
new_folder_path = os.path.join(current_directory, new_folder_name)
|
||||
print(f"Creating new folder: {new_folder_name}")
|
||||
os.makedirs(new_folder_path, exist_ok=True)
|
||||
pause_debug(f"New folder created: {new_folder_path}")
|
||||
|
||||
# Copy this file to the new folder
|
||||
source_file = os.path.abspath(__file__)
|
||||
destination_file = os.path.join(new_folder_path, os.path.basename(__file__))
|
||||
with open(source_file, 'rb') as src_file:
|
||||
with open(destination_file, 'wb') as dst_file:
|
||||
dst_file.write(src_file.read())
|
||||
pause_debug(f"File copied to new folder: {destination_file}")
|
||||
|
||||
# Create the AVP & BPE folders
|
||||
avp_folder_path = os.path.join(new_folder_path, "AVP")
|
||||
bpe_folder_path = os.path.join(new_folder_path, "BPE")
|
||||
print(f"Creating AVP and BPE folders in {new_folder_name}")
|
||||
os.makedirs(avp_folder_path, exist_ok=True)
|
||||
os.makedirs(bpe_folder_path, exist_ok=True)
|
||||
pause_debug(f"AVP and BPE folders created in {new_folder_name}")
|
||||
|
||||
def createNewAVP():
|
||||
# Creating AVP folders
|
||||
folders = [f for f in os.listdir(avp_directory) if os.path.isdir(os.path.join(avp_directory, f)) and f.startswith("AVP")]
|
||||
pause_debug(f"Folders found: {', '.join(folders)}")
|
||||
|
||||
if not folders:
|
||||
print("No folders found. Creating first folder: AVP01")
|
||||
os.makedirs(os.path.join(avp_directory, "AVP01"), exist_ok=True)
|
||||
pause_debug("First folder created: AVP01")
|
||||
else:
|
||||
pause_debug("Finding the latest folder")
|
||||
latest_number = max(int(f[3:]) for f in folders)
|
||||
new_number = latest_number + 1
|
||||
new_number = str(new_number).zfill(2)
|
||||
new_folder_name = f"AVP{str(new_number).zfill(2)}"
|
||||
pause_debug(f"New folder number determined: {new_number}")
|
||||
|
||||
input(f"\nWould you like to create {new_folder_name}? Press any key to continue or CTRL+C to exit...")
|
||||
pause_debug(f"User confirmed creation of new folder: {new_folder_name}")
|
||||
|
||||
new_folder_path = os.path.join(avp_directory, new_folder_name)
|
||||
print(f"Creating {new_folder_name}")
|
||||
os.makedirs(new_folder_path, exist_ok=True)
|
||||
pause_debug(f"New folder created: {new_folder_path}")
|
||||
|
||||
print(f"Copying files from {avp_directory} to {new_folder_name} :")
|
||||
|
||||
# Copy files from the last folder to the new folder
|
||||
last_folder_path = os.path.join(avp_directory, f"AVP{str(latest_number).zfill(2)}")
|
||||
for filename in os.listdir(last_folder_path):
|
||||
source_file = os.path.join(last_folder_path, filename)
|
||||
if os.path.isfile(source_file):
|
||||
new_filename = filename.replace(f"AVP{str(latest_number).zfill(2)}", new_folder_name)
|
||||
destination_file = os.path.join(new_folder_path, new_filename)
|
||||
with open(source_file, 'rb') as src_file:
|
||||
with open(destination_file, 'wb') as dst_file:
|
||||
dst_file.write(src_file.read())
|
||||
print(f"- {new_filename}")
|
||||
|
||||
# Calculate the total size of the newly created directory
|
||||
total_size = 0
|
||||
for dirpath, dirnames, filenames in os.walk(new_folder_path):
|
||||
for f in filenames:
|
||||
fp = os.path.join(dirpath, f)
|
||||
total_size += os.path.getsize(fp)
|
||||
total_size_mb = total_size / (1024 * 1024) # Convert bytes to megabytes
|
||||
print(f"Total size : {total_size_mb:.2f} MB")
|
||||
|
||||
def displayWelcomeMessage():
|
||||
ascii_art = """
|
||||
╔════════════════════════════════════════════════════════════════════════════╗
|
||||
║ __ _ ____ _ __ ║
|
||||
║ ____ ____ _____/ /_(_)___ ____ / __ \\_________ (_)__ / /______ ║
|
||||
║ / __ `/ _ \\/ ___/ __/ / __ \\/ __ \\/ /_/ / ___/ __ \\ / / _ \\/ __/ ___/ ║
|
||||
║ / /_/ / __(__ ) /_/ / /_/ / / / / ____/ / / /_/ / / / __/ /_(__ ) ║
|
||||
║ \\__, /\\___/____/\\__/_/\\____/_/ /_/_/ /_/ \\____/_/ /\\___/\\__/____/.py ║
|
||||
║ /____/ /___/ ║
|
||||
║ Your one-file tool to manage project directories ║
|
||||
╚════════════════════════════════════════════════════════════════════════════╝
|
||||
"""
|
||||
print(ascii_art)
|
||||
|
||||
# Function to display the menu
|
||||
def display_menu():
|
||||
print("Select an action:")
|
||||
print("1. Create New Project")
|
||||
print("2. Create New AVP")
|
||||
print("h. Help")
|
||||
print("q. Quit")
|
||||
|
||||
def displayHelp():
|
||||
print("This program is designed to help you manage project directories.")
|
||||
print("When you run the program, you will be presented with a menu.")
|
||||
print("You can choose to create a new project or create a new AVP.")
|
||||
print("If you are in a parent directory named 'Projets', you will be asked if you want to initiate a new project.")
|
||||
print("If you are in a project directory, you will be asked if you want to make a new AVP.")
|
||||
print("You can also press 'H' to display this help message.")
|
||||
print("You can press 'Q' to quit the program.")
|
||||
|
||||
def main():
|
||||
displayWelcomeMessage()
|
||||
|
||||
# Check if the current directory is a parent directory named "Projets", and if so, ask the user if they want to initiate a new project
|
||||
if os.path.basename(os.path.dirname(avp_directory)) == "Projets":
|
||||
pause_debug("Checked parent directory. It is named 'Projets'.")
|
||||
choice = input("You are in a parent directory named \"Projets\". Would you like to initiate a new project ? (y/n)\n").strip().upper()
|
||||
if choice == 'Y':
|
||||
createNewProject()
|
||||
|
||||
# Check if there is an AVP folder in the current directory, and if so, ask the user if they want to make a new AVP
|
||||
|
||||
elif "AVP" in os.listdir(current_directory):
|
||||
choice = input("You seem to be in a project directory. Would you like to make a new AVP ? (y/n)\n").strip().upper()
|
||||
if choice == 'Y':
|
||||
pause_debug("User confirmed creation of new AVP.")
|
||||
createNewAVP()
|
||||
else:
|
||||
print("invalid choice. Please try again.")
|
||||
|
||||
# If the user is not in a parent directory named "Projets" or an AVP directory, display the menu
|
||||
|
||||
else:
|
||||
while True:
|
||||
display_menu()
|
||||
choice = input("Enter your choice: ").strip().upper()
|
||||
if choice == '1':
|
||||
createNewProject()
|
||||
elif choice == '2':
|
||||
createNewAVP()
|
||||
elif choice == 'H':
|
||||
displayHelp()
|
||||
elif choice == 'Q':
|
||||
print("Exiting the program.")
|
||||
break
|
||||
else:
|
||||
print("Invalid choice. Please try again.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue