David Castex
fc70c9f910
Le script n'est pas fonctionnel, je segmente chaque fonctionnalité et vérifie au fur et a mesure.
312 lines
8.9 KiB
Python
312 lines
8.9 KiB
Python
## Script Banbou pour le prétraitement des données des dossiers d'installation
|
|
## de fibre optique.
|
|
|
|
import os, shutil
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# DEFINITIONS DE FONCTIONS
|
|
|
|
def formatter(chaine):
|
|
"""Formate selon nomenclature.
|
|
|
|
Formate la chaine de caractères passée en paramètre :
|
|
Enlève tous les accents français. Enlève la cédille du C.
|
|
Remplace les espaces ' ' et les traits d'union '-' par
|
|
des tirets bas '_'.
|
|
Ne traite pas pour le moment le AE et OE ligaturé.
|
|
Si plusieurs '_' se suivent, les réduire à un seul.
|
|
"""
|
|
#TODO : gérer les accents sur majuscules
|
|
#TODO : amélioration ou exercice, utiliser la méthode str.translate() et maketrans
|
|
resultat = ""
|
|
precedent = None
|
|
for c in chaine:
|
|
match c:
|
|
case "à" | "â" | "ä":
|
|
resultat+= "a"
|
|
case "é" | "è" | "ê" | "ë":
|
|
resultat+= "e"
|
|
case "î" | "ï":
|
|
resultat+= "i"
|
|
case "ô" | "ö":
|
|
resultat+= "o"
|
|
case "ù" | "û" | "ü":
|
|
resultat+= "u"
|
|
case "ÿ":
|
|
resultat+= "y"
|
|
case "ç":
|
|
resultat+= "c"
|
|
case " " | "-" | "_":
|
|
if('_' not in precedent ):
|
|
resultat+= "_"
|
|
c = "_"
|
|
case _:
|
|
resultat+= c
|
|
precedent = c
|
|
return resultat
|
|
|
|
|
|
|
|
def creer_liste(dossier):
|
|
"""
|
|
Construit une liste avec les fichiers de 'dossier'
|
|
|
|
Parcourt le dossier et ses sous-dossier, ajoute les fichiers
|
|
dans une liste de _Fichier.
|
|
Chaque _Fichier ajouté a ses attributs mis a jour.
|
|
Retourne la liste.
|
|
:param user_input: nom complet du dossier
|
|
:return: liste d'élements de type _Fichier
|
|
"""
|
|
liste = []
|
|
|
|
for dossier_courant, list_sousdossiers, list_fichiers in os.walk(dossier):
|
|
for fichier_courant in list_fichiers:
|
|
ce_Fichier = _Fichier.lire(dossier_courant, fichier_courant)
|
|
liste.append(ce_Fichier)
|
|
|
|
return liste
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# REPRÉSENTATION DES DONNÉES
|
|
|
|
class _Fichier:
|
|
|
|
|
|
def __init__(self,
|
|
nom_original="Pas de nom original",
|
|
chemin="Pas de chemin",
|
|
extension="Pas d'extension",
|
|
nom= "Pas de nom",
|
|
implication="Non-conforme",
|
|
taille=0):
|
|
self.nom_original = nom_original # - son nom original
|
|
self.chemin = chemin # - son chemin
|
|
self.extension = extension # - son extension
|
|
self.nom = nom # - son nom formaté
|
|
self.implication = implication # - son implication dans le projet
|
|
self.taille = taille
|
|
|
|
|
|
def afficher(self):
|
|
"""
|
|
Affiche dans la sortie standard les éléments du fichier
|
|
"""
|
|
print("nom :".ljust(16) + self.nom)
|
|
print("nom orig :".ljust(16) + self.nom_original)
|
|
print("ext :".ljust(16) + self.extension)
|
|
print("chemin :".ljust(16) + self.chemin)
|
|
print("implication :".ljust(16) + self.implication)
|
|
print("taille :".ljust(16) + str(self.taille) )
|
|
|
|
|
|
def lire(dossier, fichier):
|
|
"""
|
|
Lit le nom du fichier et du dossier
|
|
|
|
Construit un élement _Fichier et met à jour tous ses attributs
|
|
:param user_input: chemin absolue du dossier, nom du fichier
|
|
:return: un element _Fichier
|
|
"""
|
|
# initialiser un _Fichier
|
|
ce_Fichier = _Fichier()
|
|
# lire son chemin
|
|
ce_Fichier.chemin = dossier + "\\" + fichier
|
|
# déterminer son nom original et son extension
|
|
ce_Fichier.nom_original, ce_Fichier.extension = fichier.split(".", maxsplit=1) # maxsplit permet de spliter qu'une fois au cas ou on a plusieurs . dans le nom de fichier)
|
|
# formatter et écrire le nom
|
|
ce_Fichier.nom = formatter(ce_Fichier.nom_original)
|
|
# déterminer son implication
|
|
ce_Fichier.impliquer()
|
|
# calculer sa taille
|
|
ce_Fichier.taille = os.path.getsize(ce_Fichier.chemin)
|
|
|
|
return ce_Fichier
|
|
|
|
|
|
def impliquer(self):
|
|
"""Définir l'implication d'un fichier
|
|
|
|
Les fichiers nécessaires seront copiés dans le répertoire "Travail"
|
|
L'implication est définie en fonction de l'extension du fichier.
|
|
Les fichiers nécessaires sont les DWGs, les CSVs pour les datas.
|
|
Pour les shémas et relevés de topo :
|
|
- PDFs, DOCs, ODTs (Doc LibreOffice),
|
|
"""
|
|
match self.extension:
|
|
case "pdf" | "dwg" | "csv" | "doc" | "odt" | "PDF" | "DWG" | "CSV" | "DOC" | "ODT":
|
|
self.implication = "Necessaire"
|
|
case _:
|
|
self.implication = "A-ignorer"
|
|
|
|
|
|
|
|
|
|
|
|
class _Projet:
|
|
|
|
|
|
def __init__(self,
|
|
nom="Pas de nom",
|
|
date="Pas de date",
|
|
racine="Pas de chemin",
|
|
liste=[],
|
|
nb_fichiers=0,
|
|
taille=0,
|
|
rapport="Pas de fichier",
|
|
nb_shemas=0,
|
|
nb_releves=0,
|
|
nb_csv=0,
|
|
nb_dwgs=0):
|
|
self.nom = nom # nom du projet
|
|
self.date = date # date du traitement
|
|
self.racine = racine # chemin racine du projet
|
|
self.liste = liste # liste de _Fichier
|
|
self.nb_fichiers = nb_fichiers # nb de fichiers dans "Travail"
|
|
self.taille = taille # taille des fichiers dans "Travail"
|
|
self.rapport = rapport # chemin vers le visa
|
|
self.nb_shemas = nb_shemas # nb de plans de la mise en place
|
|
self.nb_releves = nb_releves # nb de rapports de relevés topo
|
|
self.nb_csv = nb_csv # nb de fichiers CSV
|
|
self.nb_dwgs = nb_dwgs # nb de fichiers DWG
|
|
|
|
|
|
def enraciner(self):
|
|
"""
|
|
récupère le repertoire de travail (working directory) courant
|
|
met à jour l'attribut 'racine'
|
|
"""
|
|
self.racine = os.getcwd()
|
|
print("Racine : ".ljust(16), f"{self.racine}\n")
|
|
|
|
|
|
def calculer_taille(self):
|
|
"""
|
|
calcule la taille des fichiers necessaires d'un liste
|
|
d'élements _Fichier.
|
|
Met à jour l'attribut 'taille' dans le projet
|
|
"""
|
|
taille = 0
|
|
for courant in self.liste:
|
|
if courant.implication in "Necessaire":
|
|
taille += courant.taille
|
|
self.taille = taille
|
|
print(f"Taille totale : {self.taille / 1024**2:.2} Mo.\n")
|
|
|
|
def dater_projet(self):
|
|
"""
|
|
recupère la date du jour
|
|
met à jour l'attribut 'date' du projet
|
|
"""
|
|
self.date = datetime.datetime.today().strftime('%Y%m%d')
|
|
|
|
|
|
def nommer_projet(self):
|
|
"""
|
|
met à jour l'attribut 'nom' en composant un nom.
|
|
Le nom est constitué du nom du dossier racine et de la date
|
|
courante formatté.
|
|
"""
|
|
self.nom = f"{os.path.basename(os.getcwd())}_{self.date}"
|
|
|
|
|
|
|
|
def preparer_dossier_travail(self):
|
|
"""
|
|
Créer un dossier "Travail" dans la racine du working directory et
|
|
le peuple des fichiers nécessaires
|
|
"""
|
|
travail = "Travail"
|
|
|
|
# création du dossier "Travail et dossier"
|
|
chemin = f"{self.racine}\\{travail}"
|
|
print(f"CHEMIN FABRIQUE : {chemin}")
|
|
try:
|
|
os.mkdir(chemin)
|
|
print (f'Dossier "{travail}" créé.')
|
|
except FileExistsError as erreur:
|
|
print(f'Avertissement: Le dossier "{travail}" existe déja.')
|
|
except OSError as err:
|
|
print(f"Fichier non trouvé. Surement un pb de chemin en amont.")
|
|
|
|
# peuplement du dossier Travail avec les fichiers necessaires
|
|
for fichier in self.liste:
|
|
print("TEST001")
|
|
if fichier.implication in "Necessaire":
|
|
print("TEST002")
|
|
shutil.copyfile(fichier.chemin + "\\" + fichier.nom_original, self.racine + "\\" + travail + "\\" + fichier.nom)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# TODO surcharger la fonction print native pour cet affichage
|
|
def afficher(liste):
|
|
"""
|
|
Affiche le nom des fichiers de la liste des _Fichiers
|
|
"""
|
|
for courant in liste:
|
|
print(courant.nom + "." + courant.extension)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# MAIN
|
|
|
|
racine = os.getcwd()
|
|
print("Répertoire courant : ".center(18), racine)
|
|
|
|
# --recupérer le chemin du dossier à traiter
|
|
pas_de_dossier = True
|
|
|
|
for a in os.scandir():
|
|
print("courant scandir() = ", a, " ", a.is_dir())
|
|
if a.is_dir():
|
|
#il y a un dossier
|
|
pas_de_dossier = False
|
|
print(f"Dossier à Traiter trouvé : {a.name}")
|
|
racine = a
|
|
|
|
if pas_de_dossier:
|
|
print("Pas de dossier trouvé...\nFin de programme\n")
|
|
os.system("pause")
|
|
# fin du programme
|
|
exit() |