повторять массивы в python и удалять эти файлы, если они существуют - PullRequest
Новая
       20

повторять массивы в python и удалять эти файлы, если они существуют

0 голосов
/ 04 мая 2018

У меня есть скрипт Python, как показано ниже, который читает четыре файла и дает разницу между этими файлами для каждого clientId, и он отлично работает: 

#!/usr/bin/python
import os, sys, re, json

with open(sys.argv[1]) as old_primary, open(sys.argv[2]) as new_primary, \
    open(sys.argv[3]) as old_second, open(sys.argv[4]) as new_second:

    prepare_json = lambda f: json.loads(re.sub(r'([0-9]+)=', '"\\1":', f.read())) 
    old_pr_data = prepare_json(old_primary)
    new_pr_data = prepare_json(new_primary)
    old_snd_data = prepare_json(old_second)
    new_snd_data = prepare_json(new_second)

    k = sys.argv[5]
    print('ClientId ' + k)
    print('pri=({})'.format(' '.join(map(str, set(old_pr_data[k]) ^ set(new_pr_data[k])))))
    print('snd=({})\n'.format(' '.join(map(str, set(old_snd_data[k]) ^ set(new_snd_data[k])))))

Вот как я его запускаю: 

python test.py old_primary_mapping.txt new_primary_mapping.txt old_secondary_mapping.txt new_secondary_mapping.txt 1

Постановка задачи

Это вывод, который я получаю для clientId 1 после запуска вышеуказанного скрипта: 

ClientId 1
pri=(192 196 176)
snd=(1482 1485 1491 1494)

Теперь я хочу перебрать массив pri и snd и создать строку файла, как показано ниже, и удалить эти файлы, если они существуют. У меня * регулярное выражение в файле.

Для pri удалите эти файлы: 

/primary/proc_192_for_*.log
/primary/proc_196_for_*.log
/primary/proc_176_for_*.log

Для snd удалите эти файлы: 

/secondary/proc_1482_for_*.log
/secondary/proc_1485_for_*.log
/secondary/proc_1491_for_*.log
/secondary/proc_1494_for_*.log

Мне просто нужно убедиться, что я удаляю эти файлы только то, что есть в массивах pri и snd. Можно ли это сделать в python?

1 Ответ

0 голосов
/ 04 мая 2018

Вы можете удалить эти файлы, используя glob как (не проверено): 

for dir_name, numbers in (('primary', pri), ('secondary', snd)):
    for number in numbers:
        for filename in glob.glob('/{}/proc_{}_for_*.log'.format(dir_name, number)):
             os.unlink(filename)
...