Объединить 2 файла с разными расширениями файлов

apple.yml
apple.md
orange.yml
orange.md

У меня есть файлы.md и.yml. Я хотел бы объединить их в один.md файл, если имя файла соответствует

Что было бы здесь следующим шагом?

contents = {}
file_extension1 = "yml"
file_extension2 = "md"

# Get all files and directories that are in current working directory
for file_name in os.listdir('folder/'):
#     print(file_name)

    # Use '.' so it doesn't match directories
    if file_name.endswith(('.' + file_extension1 , '.'+ file_extension2)):
        print(file_name)

1 ответ

Решение

Вот как вы можете использовать glob модуль:

from glob import glob

for f1 in glob("*.yml"): # For every yml file in the current folder
    for f2 in glob("*.md"): # For every md file in the current folder
        if f1.rsplit('.')[0] == f2.rsplit('.')[0]: # If the names match
            with open(f1, 'r') as r1, open(f2, 'r') as r2: # Open each file in read mode
                with open(f"{new}_f2", 'w') as w: # Create a new md file
                    w.write(r1.read()) # Write the contents of the yml file into it
                    w.write('\n') # Add a newline
                    w.write(r2.read()) # Write the contents of the md file into it



Этот код предназначен для непоследовательных yml а также mdфайлы. Если всеyml файлы имеют соответствующие md файл, и они находятся в папке с другими мешающими yml а также md файлов вы можете:

from glob import glob

m = sorted(glob("*.md"))
y = sorted(glob("*.yml"))

for f1, f2 in zip(m, y):
    with open(f1, 'r') as r1, open(f2, 'r') as r2:
        with open(f"new_{f1}", 'w') as w:
            w.write(f1.read())
            w.write('\n')
            w.write(f2.read())


ОБНОВИТЬ:

Чтобы ответить на комментарий ниже:

from os import listdir

files = listdir()
for f in files: # For every yml file in the current folder
    if f.endswith('.yml') and f.rsplit('.')[1]+'.md' in files:
        with open(f1, 'r') as r1, open(f.rsplit('.')[1]+'.md', 'r') as r2: # Open each file in read mode
            with open(f"{new}_f2", 'w') as w: # Create a new md file
                w.write(r1.read()) # Write the contents of the yml file into it
                w.write('\n') # Add a newline
                w.write(r2.read()) # Write the contents of the md file into it
Другие вопросы по тегам