Как получить ввод изображения с помощью argparse и передать его в функцию? - PullRequest
0 голосов
/ 07 мая 2020

Я пытаюсь собрать инструмент cli для преобразования изображений в черно-белые. Как передать в функцию путь к файлу и вернуть изображение без цвета? Это мой шаблонный код.

import argparse
import PIL 
def main():
    def black_white(src):
        img = Image.open('%s'%(src))
        img = img.convert('1')
        img.save('result.png')
    parser = argparse.ArgumentParser(description= 'Image Decolouriser')
    parser.add_argument('-black_white','-bg', help= 'converts to no color !!')

    if args.black_white:
        black_white(args.black_white())


if __name__ == __main__:
    main()

1 Ответ

0 голосов
/ 07 мая 2020
import argparse
def main():
    def black_white(src):
        img = Image.open('%s'%(src))
        img = img.convert('1')
        img.save('result.png')

    parser = argparse.ArgumentParser(description= 'Image Decolouriser')
    parser.add_argument('black_white', help= 'image to convert')
    args = parser.parse_args()

    black_white(args.black_white)

if __name__ == '__main__':
    main()

Пример звонка

my_prog image_path
...