Skip to content

Commit

Permalink
fix: updates to tag generation
Browse files Browse the repository at this point in the history
Changed how some of the tags are generated as per the comment here: m2ms/fragalysis-frontend#1482 (comment)
  • Loading branch information
kaliif committed Jul 29, 2024
1 parent 2275b00 commit eb17541
Showing 1 changed file with 11 additions and 12 deletions.
23 changes: 11 additions & 12 deletions viewer/target_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -1793,16 +1793,17 @@ def process_bundle(self):

logger.debug("data read and processed, adding tags")

canon_name_tag_map = {
k: v["centroid_res"] if "centroid_res" in v.keys() else "UNDEFINED"
for k, v in canon_sites_yaml.items()
}
# canon_name_tag_map = {
# k: v["centroid_res"] if "centroid_res" in v.keys() else "UNDEFINED"
# for k, v in canon_sites_yaml.items()
# }

# tag site observations
cat_canon = TagCategory.objects.get(category="CanonSites")
for val in canon_site_objects.values(): # pylint: disable=no-member
prefix = val.instance.canon_site_num
tag = canon_name_tag_map.get(val.versioned_key, "UNDEFINED")
# tag = canon_name_tag_map.get(val.versioned_key, "UNDEFINED")
tag = val.versioned_key
so_list = SiteObservation.objects.filter(
canon_site_conf__canon_site=val.instance
)
Expand All @@ -1821,12 +1822,10 @@ def process_bundle(self):
f"{val.instance.canon_site.canon_site_num}"
+ f"{next(numerators[val.instance.canon_site.canon_site_num])}"
)
tag = val.instance.name.split('+')[0]
# tag = val.instance.name.split('+')[0]
tag = val.instance.name
so_list = [
site_observation_objects[k].instance
for k in val.index_data["members"]
# site_observations_versioned[k]
# for k in val.index_data["members"]
site_observation_objects[k].instance for k in val.index_data["members"]
]
self._tag_observations(
tag, prefix, category=cat_conf, site_observations=so_list, hidden=True
Expand Down Expand Up @@ -1920,8 +1919,8 @@ def process_bundle(self):
f"F{val.instance.xtalform.xtalform_num}"
+ f"{val.instance.xtalform_site_num}"
)
# tag = f"{val.instance.xtalform.name} - {val.instance.xtalform_site_id}"
tag = val.instance.xtalform_site_id
# tag = val.instance.xtalform_site_id
tag = val.versioned_key
so_list = [
site_observation_objects[k].instance for k in val.index_data["residues"]
]
Expand Down

0 comments on commit eb17541

Please sign in to comment.