sdss-almanac 0.3__py3-none-any.whl → 0.3.1__py3-none-any.whl

almanac/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3"
+__version__ = "0.3.1"
 
 from .config import config, get_config_path, ConfigManager
 from .logger import logger

almanac/cli.py

@@ -217,7 +217,7 @@ def lookup(identifier, careful, **kwargs):
     from almanac.apogee import get_exposures
     from sdssdb.peewee.sdss5db.targetdb import (
         Assignment, AssignmentStatus,CartonToTarget, Target, Hole, Observatory,
-        Design, DesignToField
+        Design, DesignToField, Field
     )
     from sdssdb.peewee.sdss5db.catalogdb import SDSS_ID_flat
 
@@ -258,40 +258,52 @@ def lookup(identifier, careful, **kwargs):
         .select(
             fn.Lower(Observatory.label),
             AssignmentStatus.mjd,
+            Field.field_id,
         )
+        .distinct()
         .join(CartonToTarget)
         .join(Assignment)
         .join(AssignmentStatus)
         .switch(Assignment)
         .join(Hole)
         .join(Observatory)
+        .switch(Assignment)
+        .join(Design)
+        .join(DesignToField)
+        .join(Field)
         .where(
             Target.catalogid.in_(tuple(catalogids))
         &   (AssignmentStatus.status == 1)
         )
         .tuples()
     )
-    q = tuple(set([(obs, int(mjd)) for obs, mjd in q]))
+    fields = {}
+    for obs, mjd, field_id in q:
+        mjd = int(mjd)
+        fields.setdefault((obs, mjd), set())
+        fields[(obs, mjd)].add(field_id)
 
     console = Console()
 
     title = f"SDSS ID {sdss_id}"
 
-    # Create Rich table
-    rich_table = RichTable(title=f"{title}", title_style="bold blue", show_header=True, header_style="bold cyan")
+    rich_table = RichTable(
+        title=f"{title}",
+        title_style="bold blue",
+        show_header=True,
+        header_style="bold cyan"
+    )
 
     for field_name in ("#", "obs", "mjd", "exposure", "field", "fiber_id", "catalogid"):
         rich_table.add_column(field_name, justify="center")
 
-    fields = {}
     i = 1
     with Live(rich_table, console=console, refresh_per_second=4) as live:
-        for exposure in chain(*starmap(get_exposures, q)):
-            key = (exposure.observatory, exposure.mjd)
-            if (key not in fields or fields[key] == exposure.field_id) or careful:
+        for exposure in chain(*starmap(get_exposures, fields.keys())):
+            field_ids = fields[(exposure.observatory, exposure.mjd)]
+            if exposure.field_id in field_ids:
                 for target in exposure.targets:
                     if target.catalogid in catalogids:
-                        fields[key] = exposure.field_id
                         rich_table.add_row(*list(map(str, (
                             i,
                             exposure.observatory,

{sdss_almanac-0.3.dist-info → sdss_almanac-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sdss-almanac
-Version: 0.3
+Version: 0.3.1
 Summary: Everything we've got
 Author-email: Andy Casey <andrew.casey@monash.edu>
 License: BSD 3-Clause License

{sdss_almanac-0.3.dist-info → sdss_almanac-0.3.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
-almanac/__init__.py,sha256=IVYEfNRXAC4OOVay7i4tcf3qG76xQA4qXlHoStx8pRA,107
+almanac/__init__.py,sha256=nxyCJkPdw9Exgk2WWizLADkLFWCqT_I71kV6KSA_dRc,109
 almanac/apogee.py,sha256=Q7juc7BJdnP56-4UZ4CSg6FCULr22KHinHly2d0KMt8,10390
 almanac/catalog.py,sha256=tVwlLtVaUbSt1hoTpIPP2bMhgd4rpcUUP_zdr6uks2c,11633
-almanac/cli.py,sha256=WuAFwfMeID4HJoUdQt0w7wrmrkdUBhFP5MBdYU7UL7M,30009
+almanac/cli.py,sha256=rntf3UgIRk-UK-DHu8qf1ZnJrQwUs3Xd3nnkO2cF8iw,30198
 almanac/config.py,sha256=5oRO-mrtaCKIcqGxcrfeNHZIjHkULDRrI_Bv1CU5uxE,3259
 almanac/database.py,sha256=6YeuRSFU5zfIen7zU-nI4pN_Z_fnYkglPcYxzVF4Xqo,758
 almanac/display.py,sha256=IflM5Y3yOJEDkvyUYQe3bOFs1iEwoe3mUj8mbODsYDk,15388
@@ -20,9 +20,9 @@ almanac/etc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 almanac/etc/bad_exposures.csv,sha256=VQQoq6uXDhnnIyp6vw0cfeTaoyNqPi5dW0MwfKsDRCk,25394
 almanac/stash/data_models.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 almanac/stash/plugmap_models.py,sha256=8yxw6ZXuLP1ZdFuT7wCW_Z5JSJll4uymiIWRb6WvSOk,8973
-sdss_almanac-0.3.dist-info/licenses/LICENSE.md,sha256=_7dAUQQ5Ph_x1hcFXhi9aHBcqq9H11zco12eO4B3Cyg,1504
-sdss_almanac-0.3.dist-info/METADATA,sha256=ZGX1P4Vfs9n3Cbx0I9wdqrEKPRiyK1q8Tlb6XmLG11Y,6992
-sdss_almanac-0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sdss_almanac-0.3.dist-info/entry_points.txt,sha256=5Obfm-uaVxE3RnRm2W7dYaX_O_j5ggGvAW5UQG5AtuE,45
-sdss_almanac-0.3.dist-info/top_level.txt,sha256=sWBGH9a-2-nFB2-rGqioyigbkcz14t1u8rqAs_RYe_w,8
-sdss_almanac-0.3.dist-info/RECORD,,
+sdss_almanac-0.3.1.dist-info/licenses/LICENSE.md,sha256=_7dAUQQ5Ph_x1hcFXhi9aHBcqq9H11zco12eO4B3Cyg,1504
+sdss_almanac-0.3.1.dist-info/METADATA,sha256=Utp6VykVy7C5cuGllPXDZOy9lM4puReK7Y4txy_HMC0,6994
+sdss_almanac-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sdss_almanac-0.3.1.dist-info/entry_points.txt,sha256=5Obfm-uaVxE3RnRm2W7dYaX_O_j5ggGvAW5UQG5AtuE,45
+sdss_almanac-0.3.1.dist-info/top_level.txt,sha256=sWBGH9a-2-nFB2-rGqioyigbkcz14t1u8rqAs_RYe_w,8
+sdss_almanac-0.3.1.dist-info/RECORD,,