python - 有效地从文件中读取主版本号

Question

我正在编写一个脚本，它将通过读取文件来检查主要版本

在任何行存储版本号Major.Minor.Fix

像这样:

VERSION = 23.5.1

所以要阅读这个数字 23，我正在这样做

filePath = os.path.join(os.getcwd(), 'Makefile')
with open(filePath, 'r') as mkfh:
    for line in mkfh:
        if line.startswith('VERSION'):
            print line.replace(' ','').split('=')[-1].split('.')[0]
            break

他们获取主要版本的方法比使用替换和拆分两次更有效吗？

Answer 1

使用正则表达式:

import re

pattern = re.compile(r'VERSION\s*=\s*(\d+)')  # \s: space, \d: digits

with open('Makefile') as mkfh:
    for line in mkfh:
        matched = pattern.match(line)
        if matched:
            print matched.group(1)
            break

顺便说一句，如果您要访问当前工作目录中的文件，则不需要使用 os.path.join。