ais_utils: Select city in address via ciselnik

* Apply the same approach as with the birthplace.

Signed-off-by: mr.Shu <[email protected]>

Author: mrshu

eprihlaska/ais_utils.py

@@ -497,36 +497,10 @@ def fill_in_address(field, app, session, lists):
         with app.collect_operations() as ops:
             fields['country_button'].click()
 
-
-        # Write PSC instead of city name (TODO: hack to work with current AIS)
-        fields['city'].write(city_psc)
-        with app.collect_operations() as ops:
-            fields['city_button'].click()
-
-        if len(ops) == 1 and ops[-1].method == 'openDialog':
-            # Open selection dialogue
-            select_dlg = app.awaited_open_dialog(ops)
-
-            rows = app.d.table.all_rows()
-
-            # Let's try to find the correct row by checking the PSC
-            row_index = None
-            for idx, row in enumerate(rows):
-                print('city_psc: {}, row_value: {}'.format(city_psc, row.cells[2].value), file=sys.stderr)
-                if row.cells[2].value == city_psc:
-                    row_index = idx
-
-            # If we did find a row, let's select it
-            if row_index is not None:
-                app.d.table.select(row_index)
-            #else:
-            #    raise Exception('Could not find city_psc {}'.format(city_psc))
-
-            with app.collect_operations() as ops:
-                app.d.enterButton.click()
-
-            # Close selection dialogue
-            select_dlg = app.awaited_close_dialog(ops)
+        select_city_by_psc_ciselnik_code(app,
+                                         city_psc,
+                                         session[field]['city'],
+                                         fields['city'], fields['city_button'])
 
         # TODO: hack to work with current AIS
         # Write the actual city to "Post office" field
@@ -771,6 +745,9 @@ def select_city_by_psc_ciselnik_code(app, city_psc, ciselnik_code, city_field, c
     with app.collect_operations() as ops:
         city_button.click()
 
+    if len(ops) == 0:
+        return
+
     if len(ops) == 1 and ops[-1].method == 'openDialog':
         # Open selection dialogue
         select_dlg = app.awaited_open_dialog(ops)