Преглед на файлове

给标高检查的返回结果添加标高高度信息

jxing преди 5 години
родител
ревизия
25237b8859
променени са 2 файла, в които са добавени 6 реда и са изтрити 1 реда
  1. 4 0
      src/level/check_level.py
  2. 2 1
      src/level/test.py

+ 4 - 0
src/level/check_level.py

@@ -91,9 +91,11 @@ def check_level_reasonable(model_list, level_data):
     last_model_id = ''
     error = set()
     group = 0
+    elevations = {}
     for row in sorted_arr:
         current_height = row.get('elevation')
         current_model_id = row.get('model_id')
+        elevations[current_model_id] = current_height
         if current_height > last_height:
             last_height = current_height
             last_model_id = current_model_id
@@ -115,6 +117,8 @@ def check_level_reasonable(model_list, level_data):
         else:
             floor_row['has_error'] = False
             floor_row['error'] = None
+        if model_id in elevations:
+            floor_row['elevation'] = elevations[model_id]
     if len(error) == 0:
         return True
     return False

+ 2 - 1
src/level/test.py

@@ -85,4 +85,5 @@ if __name__ == '__main__':
         single_model_level = [dict(zip(level_keys, item)) for item in single_model_level]
         level_data[item.get('fid')] = single_model_level
 
-    print(check_level_reasonable(model_list, level_data))
+    print(check_level_reasonable(model_list, level_data))
+    print(model_list)