o DjJ@stddlmZddlmZddlmZddlmZddl m Z m Z ddl Z ddl Z ddlZddlZGdddeZdS))Command) BaseCommand)DEFAULT_DB_ALIAS) BACKUP_ROOT MEDIA_ROOTNc@s$eZdZddZddZddZdS)rc CsL|jdtjtddd|jd|jdtjtd}t tjt|d'}t |d }|t | Wdn1sHwYWdn1sWwYzCzt j|td gdddd Wnty}z|jd t|WYd}~nd}~ww|jd Wt|dSWt|dSt|w)Nu+Début du nettoyage de la base de données.F)database verbosity interactiveu*Nettoyage de la base de données terminé.u;début insertion de la sauvegarde dans la base de données.z e-colle.jsonrbwbjson)rformatexcludeignore app_labelr u+Erreur lors de l'insertion des données: {}u>Insertion de la sauvegarde dans la base de données terminée.)stdoutwriteFlushhandlerospathjoinrbz2open decompressreadLoaddata Exceptionrstrremove)selffichier fichier_json fichier_bz2 fichierjsoner'=/home/reyt/e-colle-new/accueil/management/commands/restore.py gere_json s.      "zCommand.gere_jsoncCs*|dkr2tt}dd|D}dd|D}|D]}t|D]}|dkr0ttj||q q|dvr|jdzQz-t tjt |d}| tz t d d d tgWntyh|jd YnwWntyy|jd Ynw|jdW|dSW|dS|wdS)u=vide la bdd puis y insère les données de la sauvegarde json2cSsg|] }tjt|qSr')rrrr).0filer'r'r( $sz&Command.gere_media..cSsg|] }tj|r|qSr')rrisdir)r+dr'r'r(r-%sz .gitignore23u&Début restauration des fichiers mediazr:xzchgrpz-Rwebu:Problème dans l'attribution des droits des fichiers mediaz1Erreur lors de la restauration des fichiers mediau)Restauration des fichiers media terminéeN)rlistdirrr rrrrtarfilerr extractall subprocessrunrclose)r! fichier_mediachoix repertoires repertoire fichiermedia archive_zipr'r'r( gere_media s:      zCommand.gere_mediac OstddttD}|s|jddSttdt|d|D]\}}|jd ||q"t d}|dksn| rLdt |krKt|ksnn|jd t d}|dksn| rLdt |krlt|krLnqL|dkrz|jd dSt |}||d}| d d d d}d |}d}tjtjt|r|jdt }|dvr|jdt }|dvs|dkr|jd n||||dkr||dSdS)NcSs g|] }|dddkr|qS)Nz.bz2r')r+fr'r'r(r-<s z"Command.handle..ucAucune sauvegarde des données n'est présente, impossible de charger des données antérieuresrz{}: {}uMQuelle sauvegarde voulez-vous charger? (Taper le numéro ou a pour abandon): arunuméro inexistant, recommencezabandon_.zecolle-media_{}.tar.xz1u1Il existe une sauvegarde des fichiers media associée, que voulez-vous faire? 1. l'ignorer 2. effacer tous les fichiers media existants et mettre ceux de la sauvegarde à la place 3. conserver les fichiers media existants et ajouter en plus ceux de la sauvegarde (en écrasant les doublons) a. abandonner a123zchoix invalide, recommencez)sortedrr3rrrziprangelenrinput isnumericintsplitrisfilerr?r)) r! app_labelsoptions fichiers_bz2numr"dater9r:r'r'r(r;s< . 0     zCommand.handleN)__name__ __module__ __qualname__r)r?rr'r'r'r(r s r)(django.core.management.commands.loaddatarr%django.core.management.commands.flushrdjango.core.management.baser django.dbrecolle.settingsrrrrr4r6r'r'r'r(s