Как я могу высушить этот код - PullRequest
Новая
       48

Как я могу высушить этот код

0 голосов
/ 09 ноября 2019

Я только начал изучать Python. Я создал эту функцию, но, как вы можете видеть в операторе «если», я продублировал код. Я только изменил file_id = filename.photo[-1].file_id на file_id = filename.video.file_id, чтобы заставить функцию работать, и она работает, нокак я могу сократить этот код

Спасибо за ваше время 

def create_post(filename):
    if filename.content_type == 'photo':
        file_id = filename.photo[-1].file_id
        file = bot.get_file(file_id)
        downloaded_file = bot.download_file(file.file_path)
        with open("image.jpg", 'wb') as new_file:
            new_file.write(downloaded_file)
        image = 'image.jpg'
        token = store_file_temporary(image)
        headers = {
            "content-type": "multipart/form-data",
            "Accept": "application/json",
            "Content-Type": "application/json"
        }
        filesd = {
            "title": randomString(),
            "safety": 'safe',
            "contentToken": token
        }
        r = requests.post(url=url + "/posts/", json=filesd, auth=(username, password),
                          headers=headers)
        print(r.json())
        return r.json()
    elif filename.content_type == 'video':
        file_id = filename.video.file_id
        file = bot.get_file(file_id)
        downloaded_file = bot.download_file(file.file_path)
        with open("image.jpg", 'wb') as new_file:
            new_file.write(downloaded_file)
        image = 'image.jpg'
        token = store_file_temporary(image)
        headers = {
            "content-type": "multipart/form-data",
            "Accept": "application/json",
            "Content-Type": "application/json"
        }
        filesd = {
            "title": randomString(),
            "safety": 'safe',
            "contentToken": token
        }
        r = requests.post(url=url + "/posts/", json=filesd, auth=(username, password),
                          headers=headers)
        print(r.json())
        return r.json()

1 Ответ

2 голосов
/ 09 ноября 2019

Как только вы получите file_id = filename.photo[-1].file_id, это в основном точно такая же обработка. Так что просто поместите эту строку в if else, а для остальных - метод. 

if filename.content_type == 'photo':
        file_id = filename.photo[-1].file_id
elif filename.content_type == 'video':
        file_id = filename.video.file_id
//call some method with file_id where you do the rest of the work.
somemeaningful_function_name(file_id)


def somemeaningful_function_name(file_id):
    //work with file_id

Вы сами говорили о принципе СУХОЙ. При приближении к таким вещам просто найдите повторяющуюся часть. Как правило, если вы повторно используете один и тот же блок кода несколько раз, просто поместите его в функцию и используйте его.

...