reformatted with PEP8 style

Author: danasmera

get_mx_records.py

@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 '''Usage: scriptname domain
-   Output will the list of MX records sorted by MX, lowest MX value first, and highest Mx value last.
-   This script uses PriorityQueue...Class was taken from Python Cookbook 3rd edition.
+   Output will list of MX records of domain sorted by MX,
+   lowest MX value first, and highest Mx value last.
+   This script uses PriorityQueue Class and is based on
+   Python Cookbook 3rd edition.
 '''
 
 __author__ = "Daniel T."
@@ -10,46 +12,50 @@
 __maintainer__ = "danasmera"
 __email__ = "[email protected]"
 
-import os,sys
+import os
+import sys
 import heapq
 
-maildomain=sys.argv[1]
+maildomain = sys.argv[1]
 # Use dig command
-mycommand="dig +short " + maildomain + " mx"
+mycommand = "dig +short " + maildomain + " mx"
+
 
 class PriorityQueue:
-  def __init__(self):
-    self._queue=[]
-    self._index=0
+    def __init__(self):
+        self._queue = []
+        self._index = 0
+
+    def push(self, item, priority):
+        heapq.heappush(self._queue, (priority, self._index, item))
+        self._index += 1
 
-  def push(self, item, priority):
-    heapq.heappush(self._queue, (priority, self._index, item))
-    self._index+=1
+    def pop(self):
+        return heapq.heappop(self._queue)[-1]
 
-  def pop(self):
-    return heapq.heappop(self._queue)[-1]
+q = PriorityQueue()
 
-q=PriorityQueue()
 
 def push_mail_servers():
-  counter=0
-  with os.popen(mycommand) as fp:
-    for line in fp:
-      mail_servers = line.strip()
-      priority, mail_server = mail_servers.split()
-      q.push(mail_server, int(priority))
-      counter+=1
+    counter = 0
+    with os.popen(mycommand) as fp:
+        for line in fp:
+            mail_servers = line.strip()
+            priority, mail_server = mail_servers.split()
+            q.push(mail_server, int(priority))
+            counter += 1
     return counter
 
+
 def main():
-  total_mx=push_mail_servers()
-  while (total_mx > 0):
-    print q.pop()
-    total_mx-=1
+    total_mx = push_mail_servers()
+    while (total_mx > 0):
+        print q.pop()
+        total_mx -= 1
 
 # Main
-if __name__=='__main__':
-  main()
+if __name__ == '__main__':
+    main()
 
 '''Usage example
 [daniel@danasmera tmp]$ dig gmail.com mx +short