Как сохранить файл (OS, Python)?

Question

[Rod_99]

Задача: сохранить по status_code файлы с их расширением в три папки (good,bad,general ).

Я изначально пробовал через такую схему:

while True:
        if():
            if(r.status_code == 200):
                with open('\\Users\\user\\Desktop\\good\\gg.txt', 'wb') as f:               
                    f.write(r.content)                                                     
        if(r.status_code != 200):
            open(r'\Users\user\Desktop\bad\gg.zip', 'wb' ).write(r.content)
        break
    open(r'\Users\user\Desktop\general\gg.zip', 'wb').write(r.content)

Но уперся в то , что в функции open нужно указывать расширение файла, а так как я скачиваю сразу с нескольких url файлы с разными расширениями этот вариант не подошёл.
Тогда я попробовал OS , вопрос в том , как сохранить файлы в папки при этом сохранить их расширение ?

Сам код :

import requests
import os

def save_file(link):
    filename = link.split('/')[-1]
   
    data = requests.get('https://ipinfo.io/json')
    print(data.text)
    r =requests.get(link,allow_redirects=True)
    print(r.status_code)
    file_name, file_ext = os.path.splitext(os.path.basename(link))
    download_name = f'gp{file_name}'
    download_dir = 'good' if r.status_code == 200 else 'bad'
    download_path = os.path.join(r'\Users\user\Desktop', download_dir,download_name)
    #f = open()
    
link1 ='https://$$$$$$$$$$$.txt'
link2 ='https://$$$$$$$$$$$.jpeg'
link3 ='https://$$$$$$$$$$$.rules'
link4 ='https://$$$$$$$$$$$.php'

save_file(link1)
save_file(link2)
save_file(link3)
save_file(link4)

Благодарен за любую помощь !

Answer 1

[Иван Четчасов]

пробовали list из либы os?

Comments

[Rod_99]

Здравствуйте, Иван.
Нет не пробовал, в OS я искал замену своему первому варианту (While Tru...).
А как list может мне помощь с сохранением фалов ?

[Иван Четчасов]

Rod_99, с его помощью вы можете получить список всех файлов в папке, понять какие - новые, какие - не очень новые, ну и сделать с ними чего вы там хотели

[Иван Четчасов]

import os

children = os.listdir(papka)
files = []

for child in children:
    if os.path.isfile(papka+'/'+child):
        files.append(papka+'/'+child)

for file in files:
    if file not in checked_files:
        ...