[PYTHON] Correspondance de fichiers

Script de correspondance pour le fichier A et le fichier B Extraire uniquement les enregistrements correspondant à la clé du fichier B

• C'est presque un mémo personnel.

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

#Correspondance de clé entre le fichier A et le fichier B
#supposition
#-Les clés du fichier A et du fichier B sont triées par ordre croissant.
#・ Le fichier a une longueur fixe
#

#Nom de fichier correspondant, nom de fichier de sortie
fileA = "a.txt"
fileB = "b.txt"
fileOut = "out.txt"

#Taille d'enregistrement de fichier
recsizeA = 4
recsizeB = 8

#Position de départ clé
keyposA = 0
keyposB = 0

#Longueur de clé
keysize = 2


#-----------------------------------------------
#Commencer le traitement
#-----------------------------------------------
fpa = open(fileA, 'rb')
fpb = open(fileB, 'rb')
fpout = open(fileOut, 'wb')

reca = fpa.read(recsizeA)
recb = fpb.read(recsizeB)

#Répétez jusqu'à ce que l'un des fichiers devienne illisible
while len(reca) == recsizeA and len(recb) == recsizeB :
    if reca[keyposA:keysize] == recb[keyposB:keysize] :
        fpout.write(recb)
        recb = fpb.read(recsizeB)
    elif  reca[keyposA:keysize] < recb[keyposB:keysize] :
        reca = fpa.read(recsizeA)
    elif  reca[keyposA:keysize] > recb[keyposB:keysize] :
        recb = fpb.read(recsizeB)

fpa.close()
fpb.close()
fpout.close()