python - 打开从 os.listdir() 找到的文件并比较里面的行？

Question

好的，我正在编写一个程序来帮助连接到无线网络。我已经记下了大部分内容(事实上，它已经完成了。我只是在处理额外的功能。)

我正在为 Arch Linux 操作系统编写名为 NetCTL 的无线网络连接后端的 GUI 前端。基本上，人们可以手动创建配置文件并将其命名为他们想要的任何名称(即“asdfasdfasdf”)，但我的将始终生成 $NetworkSSID_wifiz。

但是，每个文件中都有一行可以确定它是否用于同一网络。

行是:

ESSID='$NetworkSSID'

那么我将如何打开出现在 os.listdir 中的每个文件并检查这两个文件是否具有相同的行(最好不要产生太多开销。)？

无论是我的程序还是用户生成的所有配置文件都保存在/etc/netctl 中。

示例文件:

用户创建:

Description='A simple WPA encrypted wireless connection'
Interface=wlp2s0
Connection=wireless
Security=wpa

IP=dhcp

ESSID='MomAndKids'
# Prepend hexadecimal keys with \"
# If your key starts with ", write it as '""<key>"'
# See also: the section on special quoting rules in netctl.profile(5)
Key='########'
# Uncomment this if your ssid is hidden
#Hidden=yes

由我的程序创建:

Description='A profile generated by WiFiz for MomAndKids'
Interface=wlp2s0
Connection=wireless
Security=wpa
ESSID='MomAndKids'
Key='#######'
IP=dhcp

示例 os.listdir 输出:

['hooks', 'interfaces', 'examples', 'ddwrt', 'MomAndKids_wifiz', 'backups', 'MomAndKids']

Answer 1

这应该适合你:

from glob import glob
from os import path

config_dir = '/etc/netctl'

profiles = dict((i, {'full_path': v, 'ESSID': None, 'matches': []}) for (i, v) in enumerate(glob(config_dir + '/*')) if path.isfile(v))

for K, V in profiles.items():
    with open(V['full_path']) as f:
        for line in f:
            if line.startswith('ESSID'):
                V['ESSID'] = line.split('=',1)[1].strip()
                break # no need to keep reading.
    for k, v in profiles.items():
        if K == k or k in V['matches'] or not v['ESSID']:
            continue
        if V['ESSID'] == v['ESSID']:
            V['matches'].append(k)
            v['matches'].append(K)

for k, v in profiles.items():
    print k, v