From f585aa4fa716e2bb0c5b219ed3f5d8304b0b5edb Mon Sep 17 00:00:00 2001 From: Tim Lister Date: Fri, 14 May 2021 10:36:40 -0700 Subject: [PATCH] Add a little better attempt to find the right Yarkovsky object when multiple Body's exist --- .../management/commands/compute_long_term_scheduling.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/neoexchange/core/management/commands/compute_long_term_scheduling.py b/neoexchange/core/management/commands/compute_long_term_scheduling.py index fb859a459..4b2d743d0 100644 --- a/neoexchange/core/management/commands/compute_long_term_scheduling.py +++ b/neoexchange/core/management/commands/compute_long_term_scheduling.py @@ -54,7 +54,11 @@ def handle(self, *args, **options): self.stdout.write("\n".join(target_list)) self.stdout.write("========================") for obj_id in target_list: - orbelems = model_to_dict(Body.objects.get(name=obj_id)) + try: + target = Body.objects.get(name=obj_id) + except Body.MultipleObjectsReturned: + target = Body.objects.get(name=obj_id, active=True) + orbelems = model_to_dict(target) visible_dates, emp_visible_dates, dark_and_up_time_all, max_alt_all = monitor_long_term_scheduling(options['site_code'], orbelems, datetime.strptime(options['start_date'], '%Y-%m-%d'), options['date_range'], options['dark_and_up_time_limit']) self.stdout.write("Reading target %s" % obj_id) self.stdout.write("Visible dates:")