22from numpy import True_
33import pandas as pd
44from typing import Tuple
5- from network_wrangler import ProjectCard
5+
6+ from pathlib import Path
7+
8+ from projectcard import ProjectCard
9+ from projectcard import write_card
610
711
812def add_cards_to_registry (
@@ -30,8 +34,8 @@ def add_cards_to_registry(
3034 for card , filename in card_file_list :
3135 if card .project not in input_df ["project_added" ].values :
3236 for change_index , change_dict in enumerate (card .changes ):
33- if change_dict . get ( "category" , "Missing" ) == "Add New Roadway" :
34- if "nodes" in change_dict :
37+ if "roadway_addition" in change_dict :
38+ if "nodes" in change_dict [ "roadway_addition" ] :
3539 node_df , node_update , card = _update_registry (
3640 "nodes" ,
3741 out_df ,
@@ -52,10 +56,10 @@ def add_cards_to_registry(
5256 )
5357
5458 if node_df is not None :
55- out_df = out_df . append ( node_df , ignore_index = True )
59+ out_df = pd . concat ([ out_df , node_df ] , ignore_index = True )
5660
5761 out_df = (
58- out_df . append ( link_df , ignore_index = True )
62+ pd . concat ([ out_df , link_df ] , ignore_index = True )
5963 .drop_duplicates ()
6064 .reset_index (drop = True )
6165 )
@@ -66,7 +70,7 @@ def add_cards_to_registry(
6670 card .__dict__ .pop ("file" )
6771 if "valid" in card .__dict__ :
6872 card .__dict__ .pop ("valid" )
69- card . write ( filename = filename )
73+ write_card ( card , filename = Path ( filename ) )
7074
7175 return out_df
7276
@@ -238,7 +242,7 @@ def _update_registry(
238242
239243 subject_df = input_df [input_df ["type" ] == subject_word ]
240244
241- for subject_index , subject in enumerate (card .changes [change_index ][nodes_or_links ]):
245+ for subject_index , subject in enumerate (card .changes [change_index ]["roadway_addition" ][ nodes_or_links ]):
242246 new_id = subject [subject_id_word ]
243247
244248 _is_id_in_allowable_range (subject_word , card .project , new_id , range_in_use )
@@ -251,7 +255,7 @@ def _update_registry(
251255 "project_added" : [card .project ],
252256 }
253257 )
254- subject_df = subject_df . append ( updates_df )
258+ subject_df = pd . concat ([ subject_df , updates_df ] )
255259 else :
256260 number = _find_available_id (
257261 subject_word ,
@@ -260,23 +264,23 @@ def _update_registry(
260264 range_in_use ,
261265 subject_df ,
262266 )
263- card .changes [change_index ][nodes_or_links ][subject_index ][
267+ card .changes [change_index ]["roadway_addition" ][ nodes_or_links ][subject_index ][
264268 subject_id_word
265269 ] = number
266270 if nodes_or_links == "nodes" :
267- for i in range (0 , len (card .changes [change_index ]["links" ])):
268- if card .changes [change_index ]["links" ][i ]["A" ] == new_id :
269- card .changes [change_index ]["links" ][i ]["A" ] = number
270- if card .changes [change_index ]["links" ][i ]["B" ] == new_id :
271- card .changes [change_index ]["links" ][i ]["B" ] = number
271+ for i in range (0 , len (card .changes [change_index ]["roadway_addition" ][ " links" ])):
272+ if card .changes [change_index ]["roadway_addition" ][ " links" ][i ]["A" ] == new_id :
273+ card .changes [change_index ]["roadway_addition" ][ " links" ][i ]["A" ] = number
274+ if card .changes [change_index ]["roadway_addition" ][ " links" ][i ]["B" ] == new_id :
275+ card .changes [change_index ]["roadway_addition" ][ " links" ][i ]["B" ] = number
272276 updates_df = pd .DataFrame (
273277 {
274278 "type" : subject_word ,
275279 "id" : [number ],
276280 "project_added" : [card .project ],
277281 }
278282 )
279- subject_df = subject_df . append ( updates_df )
283+ subject_df = pd . concat ([ subject_df , updates_df ] )
280284 write_updated_card = True
281285
282286 return subject_df , write_updated_card , card
0 commit comments