python - 如何实现在接收文件名时增加文件名的函数？

Question

如何编写一个函数来为传入的文件创建一个增量名称？

def callback(ch,method, properties, body):

    Payload = body.decode("utf-8")
    Payload = ast.literal_eval(Payload)

    with open("received.png", "wb") as f:
        f.write(Payload)

    print(type(Payload))
    print("Data Received : {}".format(Payload))

在这里，我收到一个 image.png 并将其命名为 received.png。我想编写一个函数来自动重命名下一个传入的图像，比如 received_2.png

标签： pythonfilenamesincrement

Answer 1

您可以获取当前目录中的文件数并将其加 1。这是获取当前目录中文件数的代码

import os, os.path

print(len([name for name in os.listdir('.') if os.path.isfile(name)]))