Home  >  Article  >  Backend Development  >  Batch modification of python file names

Batch modification of python file names

巴扎黑
巴扎黑Original
2017-09-11 10:42:371546browse

This article shares with you a python script that implements batch modification of file names in just 10 lines of code. It is recommended to everyone. Friends in need can refer to it

I once thought that writing scripts is It was difficult until I met Python

Preface With the follow-up of domestic copyright awareness, many film and television music resources have begun to charge, and Dipan often blocks various resources at will. Therefore, in order to protect resources, old Drivers are increasingly inclined to download resources locally, but as the amount of local resources increases, resource management becomes a headache. There are many ways to manage resources. I prefer to name a resource. An easy-to-remember name, but changing file names is a boring and time-consuming task, so a batch rename script becomes very important.

1. Batch prefix function


Batch rename

Batch rename logic, divided into three steps:

( 1) Determine the files that need to be named
(2) Determine the new file name
(3) Change the name

Once the logic is clarified, coding becomes very simple


10 lines of code

Source code for the above picture:


# 导入python内置的os模块和sys模块
import os
import sys

# 程序入口
if __name__ == "__main__":
  # 获取需要添加的前缀
  pre = input("请输入需要添加的前缀:")
  # 为了美观,为前缀添加一个中括号
  mark = "[%s]"%pre
  # 获取本目录下所有的文件名
  old_names = os.listdir()
  # 遍历目录下的文件名
  for old_name in old_names:
    # 跳过本脚本文件
    if old_name != sys.argv[0]:
      # 用新的文件名替换旧的文件名
      os.rename(old_name, mark+old_name)

2. (Extended) Add and remove prefix function:

The 10-line program above can only add prefixes in batches, but cannot remove them in batches. It is not perfect. This is the source code that adds the function of removing prefixes.

Add/Remove

Above picture Source code:


import os
import re
import sys

def add_mark():
  pre = input("请输入需要添加的前缀:")
  mark = "[%s]"%pre
  old_names= os.listdir()
  for old_name in old_names:
    if old_name != sys.argv[0]:
      os.rename(old_name, mark+old_name)

def remove_mark():
  old_names= os.listdir()
  for old_name in old_names:
    try:
      result = re.match(r"(^\[.*\])(.*)", old_name).group(2)
      rm = old_name

      if result:
        os.rename(old_name, result)
      print("已为%s移除前缀"%rm)
    except Exception as e:
      pass

def main():
  while True:
    option = int(input("请选择功能数值:\n1.添加前缀\n2.删除前缀\n3.退出程序\n"))
    if option == 1:
      add_mark()
    elif option == 2:
      remove_mark()
    else:
      exit()

if __name__ == "__main__":
  main(

The above is the detailed content of Batch modification of python file names. For more information, please follow other related articles on the PHP Chinese website!

Statement:
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn