@@ -116,20 +116,23 @@ def replace_dev_reqs(file, pkg_root):
116
116
adjusted_req_lines = []
117
117
118
118
with open (file , "r" ) as f :
119
- for line in f :
120
- args = [part .strip () for part in line .split () if part and not part .strip () == "-e" ]
121
- amended_line = " " .join (args )
119
+ original_req_lines = list (line .strip () for line in f )
122
120
123
- if amended_line .endswith ("]" ):
124
- trim_amount = amended_line [::- 1 ].index ("[" ) + 1
125
- amended_line = amended_line [0 : (len (amended_line ) - trim_amount )]
121
+ for line in original_req_lines :
122
+ args = [part .strip () for part in line .split () if part and not part .strip () == "-e" ]
123
+ amended_line = " " .join (args )
124
+ extras = ""
126
125
127
- adjusted_req_lines .append (amended_line )
126
+ if amended_line .endswith ("]" ):
127
+ amended_line , extras = amended_line .rsplit ("[" , maxsplit = 1 )
128
+ if extras :
129
+ extras = f"[{ extras } "
130
+
131
+ adjusted_req_lines .append (f"{ build_whl_for_req (amended_line , pkg_root )} { extras } " )
128
132
129
133
req_file_name = os .path .basename (file )
130
- logging .info ("Old {0}:{1}" .format (req_file_name , adjusted_req_lines ))
134
+ logging .info ("Old {0}:{1}" .format (req_file_name , original_req_lines ))
131
135
132
- adjusted_req_lines = list (map (lambda x : build_whl_for_req (x , pkg_root ), adjusted_req_lines ))
133
136
logging .info ("New {0}:{1}" .format (req_file_name , adjusted_req_lines ))
134
137
135
138
with open (file , "w" ) as f :
0 commit comments