update: new table format (retrocompatible), bug correction and plotting ability

Source/NGC_v1.py

@@ -0,0 +1,107 @@
+import os
+from astroquery.vizier import Vizier
+from astrobs_v1_toolbox import read_cfg, create_table
+import astropy.coordinates as coord
+import astropy.units as u
+
+OUT_DIR = "./Output/"
+QUIT = ["QUIT", "EXIT", "Q"]
+
+obj_type_list = """\t\033[32m1. \033[0mOpen Cluster
+\t\033[32m2. \033[0mGlobular Cluster
+\t\033[32m3. \033[0mDiffuse Nebula
+\t\033[32m4. \033[0mPlanetary Nebula 9 Object in Small Magellanic Cloud
+\t\033[32m5. \033[0mGalaxy
+\t\033[32m6. \033[0mCluster associated with nebulosity
+\t\033[32m7. \033[0mNon existent
+\t\033[32m8. \033[0mObject in Large Magellanic Cloud
+\t\033[32m0. \033[0mUnverified southern object
+\t\033[32m(blank). \033[0mAny"""
+
+
+def get_NGC(filename: str,
+            obj_type: int = None,
+            directory: str = OUT_DIR,
+            extension: str = ".cfg"):
+    if obj_type == None:
+        obj_type = ">0"
+
+    Vizier.clear_cache()
+    config = read_cfg(filename, directory, extension)
+    date = config["SUN_SET"][:10]
+    loc = config["LOCATION"]
+
+    NGC = Vizier(catalog="VII/1B", 
+                 columns=["NGC", "Type", "_RAJ2000", "_DEJ2000", "+Mag"])
+    objects = NGC.query_constraints(RA1975 = config["RA_CONST"],
+                                    DE1975 = config["DE_CONST"],
+                                    Mag = "<999",
+                                    Type = str(obj_type))[0]
+    table = create_table()
+    for i in range(len(objects)):
+        mag = "Mag: {}".format(objects["Mag"][i])
+        ra = "{}".format(coord.Angle(objects["_RAJ2000"][i], 
+                         unit=u.degree).to_string(u.hourangle,
+                                                  sep=":", 
+                                                  pad=True))
+        dec =  "{}".format(coord.Angle(objects["_DEJ2000"][i], 
+                           unit=u.degree).to_string(u.degree, 
+                                                    alwayssign=True, 
+                                                    sep=":",
+                                                    pad=True))
+        name = "NGC {}".format(objects["NGC"][i])
+        table.add_row({"seq": 0, 
+                       "name": name, 
+                       "main_id": name,
+                       "ra": ra, 
+                       "dec": dec,
+                       "notes": mag})
+    table.write("{}{}_{}_ngc.xml".format(directory, date, loc), 
+                  format="votable",
+                  overwrite=True)
+    table.pprint()
+    return None
+
+def main(directory: str = OUT_DIR,
+         extension: str = ".cfg") -> tuple:
+    filelist = [fname for fname in os.listdir(directory) if extension in fname]
+    print("\033[32m"
+          + "Select a configuration"
+          + "\033[0m")
+    for i in range(len(filelist)):
+        print("\033[32m"
+              + "\t{}. ".format(i+1)
+              + "\033[0m"
+              + "{}".format(filelist[i][:-4]))
+    try: 
+        answer = input("\033[32m" + "Choice: " + "\033[0m")
+        if answer in QUIT:
+            return 0
+        selection = int(answer) - 1
+        filename = filelist[selection][:-4]
+    except Exception as e:
+        print("\033[91m"
+              + ("Error! Input value not recognized, "
+                 "selected the last location instead")
+              + "\033[0m")
+        selection = -1
+        filename = filelist[selction][:-4]
+    print("\033[32m"
+          + "Select a type of object"
+          + "\033[0m")
+    print("\033[32m" + obj_type_list + "\033[32m")
+    try:
+        answer = input("\033[32m" + "Choice: " + "\033[0m")
+        if answer in QUIT:
+            return 0
+        if answer in [" ", ""]:
+            obj_type = None
+        obj_type = int(answer)
+    except Exception as e:
+        obj_type = None
+
+    get_NGC(filename, obj_type)
+    return 0
+
+if __name__ == "__main__":
+    main()