File size: 2,154 Bytes
ad2000a 48dfeff ad2000a 48dfeff 46c2ede 48dfeff ad2000a 48dfeff ad2000a 48dfeff ad2000a 5d13184 ad2000a 48dfeff ad2000a 5d13184 ad2000a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
import hashlib
import csv
import os
def hash_md5(filepath):
md5_hash = hashlib.md5()
try:
with open(filepath, "rb") as file:
for chunk in iter(lambda: file.read(128 * md5_hash.block_size), b''):
md5_hash.update(chunk)
return md5_hash.hexdigest()
except Exception as e:
return str(e)
CSV_FILE_PATH = "audio_plus_hash_uniq_07102024.csv"
def update_csv_with_files(csv_file_path, audio_old_path, audio_16000_path, new_transcription):
hash_old = hash_md5(audio_old_path)
hash_new = hash_md5(audio_16000_path)
update_csv(csv_file_path, hash_old, hash_new, audio_old_path, new_transcription)
def update_csv(csv_file_path, search_hash, hash_16000, new_path, new_transcription):
# Use read/write mode to modify the relevant line or append if not found
with open(csv_file_path, mode='r+', newline='', encoding='utf-8') as file:
reader = csv.DictReader(file)
fieldnames = reader.fieldnames
rows = list(reader)
found = False
# Locate the row with the matching hash
for i, row in enumerate(rows):
if row['hash'] == search_hash or row['hash_16000'] == search_hash:
rows[i]['hash_16000'] = hash_16000
rows[i]['transcription'] = new_transcription
found = True
break
if found:
# Move file pointer to the beginning and write only the updated row
file.seek(0) # Go to the beginning of the file
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader() # Ensure the header is written
writer.writerows(rows) # Write all rows back, with the updated one
return
# Append a new row if the hash is not found
with open(csv_file_path, mode='a', newline='', encoding='utf-8') as append_file:
writer = csv.DictWriter(append_file, fieldnames=fieldnames)
writer.writerow({
'hash': search_hash,
'hash_16000': hash_16000,
'filepath': new_path,
'transcription': new_transcription
})
|