如何执行python程序并自动以不同的文件名保存输出

Question

I have a Python program which processes images and save the resulting output to a file. I am receiving the input and output file names via console using input(). But I want to save the output automatically with the file name as something like "input_file_name_out". Note the string "_out" attached to the input file name

This will help me get user input only for Input file names thereby saving little time for the user since he doesn't need to think everytime what should be the output file name

Sample code snippet

if __name__ == "__main__":

    in_file = input()
    out_file = input()

    processed = process_image(in_file,out_file)

Expected code snippet

if __name__ == "__main__":

    in_file = input()
    # out_file = input() ------Not getting the file name from user thereby supplying only one argument to the below function

        processed = process_image(in_file)

# And within the function
def process_image(img):
    .
    .
    .
    out_file = img+"_out.jpg" ##### What should come here to achieve my requirement ########
    cv2.imwrite(out_file,processed_image)

Answer 1

you could make use of the Path class from the pathlib module:

from pathlib import Path

in_file = Path(input())
insert = '_out'

out_file = in_file.parent / (in_file.stem + insert + in_file.suffix)

Example:

in_file = Path('D:/folder/test.jpg')
insert = '_out'
out_file = in_file.parent / (in_file.stem + insert + in_file.suffix)

print(out_file)
D:\folder\test_out.jpg

Answer 2

All you have to do is,

• Take the input_img.jpg that you get from user input
• Split the input_img.jpg with delimiter . and that will give you this list: ['input_img', 'jpg']
• Pick up the first element and add _out.jpg suffix

def process_image(img):

    # you should split with delimiter ('.'), take the first 
    # element of the list and add "_out.jpg" suffix

    split_name = img.split('.')
    out_file = split_name[0] + "_out" + "." + split_name[1] 
    cv2.imwrite(out_file, processed_image)


if __name__ == "__main__":

    in_file = input()
    processed = process_image(in_file)

Answer 3

Well that's simple! Just take the "in_file" string you asked for, and concat "_out" to the end and pass it as the out_file.

If you only process ".jpg"

cv2.imwrite(in_file[:-4]+"_out.jpg",processed_image)

if its variable:

cv2.imwrite(in_file[:-4]+"_out"+in_file[-4:],processed_image)