Überprüfen Sie, ob das Befehlszeilenargument bereits verwendet wird

Question

Ich versuche, Reverse Lookup der IP-Adresse (Argument) zu überprüfen. und dann schreibe das Ergebnis in die txt-Datei. Wie kann ich überprüfen, ob die IP-Adresse (Argument) bereits in der Datei registriert ist? Wenn ja, muss ich das Skript verlassen.

Mein Skript:

import sys, os, re, shlex, urllib, subprocess 

cmd = 'dig -x %s @192.1.1.1' % sys.argv[1] 

proc = subprocess.Popen(shlex.split(cmd), stdout=subprocess.PIPE) 
out, err = proc.communicate() 

# Convert to list of str lines 
out = out.decode().split('\n') 

# Only write the line containing "PTR" 
with open("/tmp/test.txt", "w") as f: 
for line in out: 
    if "PTR" in line: 
     f.write(line) 

Answer 1

Etwas wie:

otherIps = [line.strip() for line in open("<path to ipFile>", 'r')] 
theIp = "192.168.1.1" 
if theIp in otherIps: 
    sys.exit(0) 

otherIps enthält eine list der IP-Adressen auf ipFile, dann müssen Sie prüfen, ob theIp bereits auf otherIps, wenn ja, Verlasse das Skript.

Answer 2

Wenn die Datei nicht zu groß ist, könnten Sie tun:

with open('file.txt','r') as f: 
    content = f.read() 
if ip in content: 
    sys.exit(0) 

Nun, wenn die Datei groß ist, und Sie möchten, um mögliche Speicherprobleme vermeiden Sie mmap wie so verwenden:

import mmap 
with open("file.txt", "r+b") as f: 
    # memory-map the file, size 0 means whole file 
    mm = mmap.mmap(f.fileno(), 0) 
    if mm.find(ip) != -1: 
     sys.exit(0) 

Die mmap.find(string[, start[, end]]) ist gut dokumentiert here.