2011-11-29 4 views
0

파일에서 읽기 위해 적응해야하는 다음 스크립트가 있는데, 구체적으로 '도메인'과 '예상 응답'을 가져와야합니다. 현재 나는 코드를 수동으로 작성했기 때문에 리팩터링 할만큼 충분히 숙련되지 않았습니다.텍스트 파일에서 읽을 스크립트 수정.

#!/usr/bin/python 
import dns.resolver 
from smtplib import SMTP 
import datetime 

debuglevel = 0 

domain = 'exampledomain.co.uk' 
expected_responses = ['example.co.uk.', 'example2.co.uk.'] 
for x in dns.resolver.query(domain, 'MX'): 
     if x.to_text().split()[1] not in expected_responses: 
       print "Unexpected MX record found!" 
       smtp = SMTP() 
       smtp.set_debuglevel(debuglevel) 
       smtp.connect('localhost', 25) 

       from_addr = "MX ERROR <[email protected]>" 
       to_addr = "[email protected]" 

       subj = "MX ERROR" 
       date = datetime.datetime.now().strftime("%d/%m/%Y %H:%M") 

       message_text = "Dearest colleagues\nSomething appears to be wrong with the MX records for example.co.uk\n\nDON'T PANIC!\n" 

       msg = "From: %s\nTo: %s\nSubject: %s\nDate: %s\n\n%s" % (from_addr, to_addr, subj, date, message_text) 


       smtp.sendmail(from_addr, to_addr, msg) 
       smtp.quit() 
     else: 
       print x.to_text().split()[1] + " example.uk MX records OK!" 

내 첫번째 생각은 .txt 인 파일을 사용했지만, 난 내 파이썬 지식과 같은 파일에서 도메인 및 예상 됨 구별하는 방법을 알아낼 수 없었다. 검색에 대한 정보를 얻었고 가져 오기 CSV 기능에 대해 읽었습니다.이 옵션은 가장 좋은 옵션으로 보이지만 사용하는 데 아무런 경험이 없습니다.

누군가 내 코드에 import csv를 적용 할 수있는 방법에 대한 예를 제공 할 수 있습니까?

감사

크리스

+1

http://docs.python.org/library/csv.html –

답변

관련 문제