Updated script to support .zst instead of .gz

This commit is contained in:
datechnoman 2024-02-02 04:07:17 +00:00
parent 10c2658bff
commit 0fd62a7391

View File

@ -9,7 +9,7 @@ def read_paths_from_file(file_path):
return [line.strip() for line in file]
def list_files_in_directory(directory):
return [f.replace("_urls.txt", "") for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))]
return [f.replace("_urls.txt.zst", ".gz") for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))]
def main():
# Get the directory containing wat.paths file
@ -54,4 +54,4 @@ def main():
print(f"List of modified files in {compare_directory} written to {output_file_path}")
if __name__ == "__main__":
main()
main()