#!/usr/bin/env python3

# -*- coding: utf-8 -*-

import sys

import subprocess

import os

import csv

from shutil import copyfile

def GetElement(file, i, j):

    with open(file, newline='') as csvfile:

        reader = csv.reader(csvfile, delimiter=',', quotechar='|')

        for line in reader:

            if reader.line_num - 1 == i:

                return line[j]

# Attention : le séparateur du fichier CSV est la virgule ","

#-------------------------------------------------------------

def NombreEtu(filename):

    with open(filename, 'r') as f:

        i = -1 # i = 0 (-1 est là pour compenser l'en-tête)

        for ligne in f:

            i += 1

    return i

if sys.argv[1] == "?" :

    print("[DossierInitial] [DossierRenommer] [DossierCrypter] [OwnerPwd] [Generic] [Denomination] [FichierCSV]")

    print("Generic : Nom générique des fichiers à renommer. Par exemple : MAT_PDFSAM_ pour MAT_PDFSAM_1.pdf, MAT_PDFSAM_2.pdf, etc.")

    print("Denomination, par exemple : DS2_MAT2")

    sys.exit()

NomDossierInitial = sys.argv[1]

NomDossierRenommer = sys.argv[2]

NomDossierCrypter = sys.argv[3]

OwnerPwd = sys.argv[4] # OwnerPass

Generic = sys.argv[5]

Denomination = sys.argv[6]

FichierCSV = sys.argv[7]

k = 0

NombreEtudiants = NombreEtu(FichierCSV)

thepath = os.getcwd()

repini = thepath+"/"+NomDossierInitial # dossier avec les fichiers initiaux (non cryptés)

reprenom = thepath+"/"+NomDossierRenommer # dossier avec les fichiers renommés (sans cryptage) 

repcrypt = thepath+"/"+NomDossierCrypter # dossier qui contiendront les fichiers cryptés

with open('Reperage_Numeros_Sujets.csv', 'w', newline='',encoding='utf-8') as csvfile:

    spamwriter = csv.writer(csvfile, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL)

    spamwriter.writerow(['Num-Sujet','Nom','Prenom','QCM-Number'])

#    spamwriter.writerow(['Spam', 'Lovely Spam', 'Wonderful Spam'])

for j in range(1,NombreEtudiants + 1) :    

    File = os.path.join(repini, Generic+str(j)+".pdf")

    #print(nfc)

    k = k + 1 # Numéro de la ligne dans le fichier csv (commence à 0)

    if k > NombreEtudiants :

        break

    #------------------------------

    NomETU = GetElement(FichierCSV,k,0)

    PrenomETU = GetElement(FichierCSV,k,1)

    ReadPwd = GetElement(FichierCSV,k,2) # C'est le QCM-Number (Identifiant étudiant)

    NomFichierRenommer = reprenom+"/"+NomETU+"_"+PrenomETU+"_"+Denomination+"_Sujet-Numero_"+str(k)+".pdf"

    NomFichierCrypter = repcrypt+"/"+NomETU+"_"+PrenomETU+"_"+Denomination+"_Sujet-Numero_"+str(k)+"_p.pdf"

    copyfile(File, NomFichierRenommer)

    # Cryptage du fichier PDF

    subprocess.call(["qpdf","--encrypt",ReadPwd,OwnerPwd,"256","--",File,NomFichierCrypter])

    with open('Reperage_Numeros_Sujets.csv', 'a', newline='',encoding='utf-8') as csvfile: # 'a' pour insertion dans csv

        spamwriter = csv.writer(csvfile, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL)

        spamwriter.writerow([str(k),NomETU,PrenomETU,ReadPwd])