support domU dns names

DrGonzo65 · Apr 25, 2012 · 9a0c32b · 9a0c32b
1 parent 3894796
commit 9a0c32b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bin/ec2who b/bin/ec2who
@@ -23,7 +23,7 @@ def get_instances_for_query(ec2_conn, query):
             filters["ip-address"] = args.query
 
     # dns
-    if re.match(r"^(ip|ec2)-", args.query):
+    if re.match(r"^(ip|ec2|domU)-", args.query):
         dns = args.query
 
         pairs = [("ec2", ".compute-1.amazonaws.com"), ("ip", ".ec2.internal")]
@@ -32,7 +32,7 @@ def get_instances_for_query(ec2_conn, query):
                 dns += ".*"
                 break
 
-        key = "private-dns-name" if dns.startswith("ip") else "dns-name"
+        key = "private-dns-name" if (dns.startswith("ip") or dns.startswith("domU")) else "dns-name"
         filters[key] = dns
 
     # instance name