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

时间:2013-05-21 08:38:27

标签: python linux

好吧,所以我正在编写一个程序来帮助连接到无线网络。我把它的大部分都放下了(事实上,它已经完成了。我只是在研究额外的功能。)

我正在为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']

2 个答案:

答案 0 :(得分:2)

这应该适合你:

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

答案 1 :(得分:0)

import os

all_essid = []

for file in os.listdir('.'):

    if not os.path.isfile(file):
        break

    with open(file) as fo:
        file_lines = fo.readlines()

    for line in file_lines:

        if line.startswith('ESSID')

            if line in all_essid:
                print 'duplicate essid %s' % line

            all_essid.append(line)

如果你想进入目录,你可以尝试os.walk;

    for root, dirs, files in os.walk("."):

        for file in files:

             # etc.