1616
1717import atexit
1818import os
19- import re
2019import shutil
2120import sys
2221from pathlib import Path
@@ -118,6 +117,7 @@ def main(
118117 absolute_path_prefix = resolved_requirements_file [
119118 : - (len (requirements_file ) - len (repository_prefix ))
120119 ]
120+
121121 # As srcs might contain references to generated files we want to
122122 # use the runfiles file first. Thus, we need to compute the relative path
123123 # from the execution root.
@@ -162,19 +162,12 @@ def main(
162162 argv .append (
163163 f"--output-file={ requirements_file_relative if UPDATE else requirements_out } "
164164 )
165- src_files = [
165+ argv . extend (
166166 (src_relative if Path (src_relative ).exists () else resolved_src )
167167 for src_relative , resolved_src in zip (srcs_relative , resolved_srcs )
168- ]
169- argv .extend (src_files )
168+ )
170169 argv .extend (extra_args )
171170
172- # Replace in the output lock file
173- # the lines like: # via -r /absolute/path/to/<requirements_file>
174- # with: # via -r <requirements_file>
175- # For Windows, we should explicitly call .as_posix() to convert \\ -> /
176- absolute_src_prefixes = [Path (src ).absolute ().parent .as_posix () + "/" for src in src_files ]
177-
178171 if UPDATE :
179172 print ("Updating " + requirements_file_relative )
180173
@@ -192,14 +185,14 @@ def main(
192185 # and we should copy the updated requirements back to the source tree.
193186 if not absolute_output_file .samefile (requirements_file_tree ):
194187 atexit .register (
195- lambda : shutil .copy (absolute_output_file , requirements_file_tree )
188+ lambda : shutil .copy (
189+ absolute_output_file , requirements_file_tree
190+ )
196191 )
197- cli (argv , standalone_mode = False )
192+ cli (argv , standalone_mode = False )
198193 requirements_file_relative_path = Path (requirements_file_relative )
199194 content = requirements_file_relative_path .read_text ()
200195 content = content .replace (absolute_path_prefix , "" )
201- for absolute_src_prefix in absolute_src_prefixes :
202- content = content .replace (absolute_src_prefix , "" )
203196 requirements_file_relative_path .write_text (content )
204197 else :
205198 # cli will exit(0) on success
@@ -221,15 +214,6 @@ def main(
221214 golden = open (_locate (bazel_runfiles , requirements_file )).readlines ()
222215 out = open (requirements_out ).readlines ()
223216 out = [line .replace (absolute_path_prefix , "" ) for line in out ]
224-
225- def replace_via_minus_r (line ):
226- if "# via -r " in line :
227- for absolute_src_prefix in absolute_src_prefixes :
228- line = line .replace (absolute_src_prefix , "" )
229- return line
230- return line
231-
232- out = [replace_via_minus_r (line ) for line in out ]
233217 if golden != out :
234218 import difflib
235219
0 commit comments