Python3 提升你脚本的逼格系列（二）

通过文件名查找文件

有时候，你不想在 Python 脚本中调用 shell，或者你要实现一些 shell 不能做的功能。

这样可以是 os.walk() 函数

#!/usr/bin/env python3

import os

def findfile(start, name):
    """
    start: 查找的路径
    name: 查找的文件名
    """
    for relpath, dirs, files in os.walk(start):
        if name in files:

            # 路径拼接
            full_path = os.path.join(start, relpath, name) 
            print(os.path.normpath(os.path.abspath(full_path)))
if __name__ == '__main__': 
        findfile(sys.argv[1], sys.argv[2])

os.walk() 方法为我们遍历目录树，每次进入一个目录，它会返回一个三元组，包 含相对于查找目录的相对路径，一个该目录下的目录名列表，以及那个目录下面的文件 名列表。

os.path.abspath() , 它接受一个路径，可能是相对路径， 最后返回绝对路径。
os.path.normpath() ，用来返回正常的标准化路径，可以解决双斜杆、对目录的多重引用的问题等。

利用上面的函数稍作修改，可以实现一个打印所有最近被修改过的文件的函数。

#!/usr/bin/env python3
import os
import time

def modified_within(top, seconds):
    now = time.time()
    for path, dirs, files in os.walk(top):
        for name in files:
            fullpath = os.path.join(path, name) 
            if os.path.exists(fullpath):
                mtime = os.path.getmtime(fullpath) 
                if mtime > (now - seconds):
                    print(fullpath)

if __name__ == '__main__': import sys
    if len(sys.argv) != 3:
        print('Usage: {} dir seconds'.format(sys.argv[0])) 
        raise SystemExit(1)
        
    modified_within(sys.argv[1], float(sys.argv[2]))

---

解析配置文件

好的软件或者服务的配置文件都是 .ini 格式的。
如何正确的解析呢？标准库中的 configparser 模块可以轻松实现。

假设有下面的 MySQL 配置文件。

[mysqld]
datadir=/var/lib/mysql
socket=/var/lib/mysql/mysql.sock
# Disabling symbolic-links is recommended to prevent assorted security risks
symbolic-links=0

[mysqld_safe]
log-error=/var/log/mariadb/mariadb.log
pid-file=/var/run/mariadb/mariadb.pid

#
# include all files from the config directory
#
!includedir /etc/my.cnf.d

操作代码

In [50]: cfg.read('my.cnf')
Out[50]: ['my.cnf']

In [51]: cfg.sections()
Out[51]: ['mysqld', 'mysqld_safe']

In [52]: cfg.options('mysqld')
Out[52]: ['datadir', 'socket', 'symbolic-links', 'port']

In [53]: cfg.get('mysqld', 'datadir')
Out[53]: '/var/lib/mysql'

In [54]: cfg.get('mysqld', 'socket')
Out[54]: '/var/lib/mysql/mysql.sock'

In [55]: cfg.set('mysqld', 'port', '3306')

In [56]: cfg.get('mysqld', 'port')
Out[56]: '3306'

In [57]: cfg.getint('mysqld', 'port')
Out[57]: 3306