Update readqubdata.py

Author: lukeshingles

artisatomic/readqubdata.py

@@ -209,16 +209,17 @@ def read_qub_levels_and_transitions(atomic_number, ion_stage, flog):
                 id_lower = int(row[1])
                 A = float(row[2])
                 if A > 2e-30:
-                    namefrom = qub_energylevels[id_upper].levelname
-                    nameto = qub_energylevels[id_lower].levelname
+                    level_upper = qub_energylevels[id_upper]
+                    level_lower = qub_energylevels[id_lower]
+                    assert level_upper is not None
+                    assert level_lower is not None
+                    namefrom = level_upper.levelname
+                    nameto = level_lower.levelname
                     # WARNING replace with correct selection rules!
-                    forbidden = qub_energylevels[id_upper].parity == qub_energylevels[id_lower].parity
+                    forbidden = level_upper.parity == level_lower.parity
                     transition_count_of_level_name[namefrom] += 1
                     transition_count_of_level_name[nameto] += 1
-                    lamdaangstrom = 1.0e8 / (
-                        qub_energylevels[id_upper].energyabovegsinpercm
-                        - qub_energylevels[id_lower].energyabovegsinpercm
-                    )
+                    lamdaangstrom = 1.0e8 / (level_upper.energyabovegsinpercm - level_lower.energyabovegsinpercm)
                     if (id_lower, id_upper) in upsilondict:
                         coll_str = upsilondict[(id_lower, id_upper)]
                     elif forbidden:
@@ -278,15 +279,16 @@ def read_qub_levels_and_transitions(atomic_number, ion_stage, flog):
                         id_lower = int(row[1])
                     A = float(row[2].replace("-", "E-").replace("+", "E+"))
                     if A > 2e-30:
-                        namefrom = qub_energylevels[id_upper].levelname
-                        nameto = qub_energylevels[id_lower].levelname
-                        forbidden = artisatomic.check_forbidden(qub_energylevels[id_upper], qub_energylevels[id_lower])
+                        level_upper = qub_energylevels[id_upper]
+                        level_lower = qub_energylevels[id_lower]
+                        assert level_upper is not None
+                        assert level_lower is not None
+                        namefrom = level_upper.levelname
+                        nameto = level_lower.levelname
+                        forbidden = artisatomic.check_forbidden(level_upper, level_lower)
                         transition_count_of_level_name[namefrom] += 1
                         transition_count_of_level_name[nameto] += 1
-                        lamdaangstrom = 1.0e8 / (
-                            qub_energylevels[id_upper].energyabovegsinpercm
-                            - qub_energylevels[id_lower].energyabovegsinpercm
-                        )
+                        lamdaangstrom = 1.0e8 / (level_upper.energyabovegsinpercm - level_lower.energyabovegsinpercm)
                         if (id_lower, id_upper) in upsilondict:
                             coll_str = upsilondict[(id_lower, id_upper)]
                         elif forbidden: