def _capture_annotated_out_reqlines(results, resolver, constraints, format_control): with NamedTemporaryFile(delete=False) as tmp_file: tmp_file_name = tmp_file.name try: writer = OutputWriter(src_files=None, dst_file=tmp_file_name, dry_run=False, emit_header=False, emit_index=False, emit_trusted_host=False, annotate=True, generate_hashes=False, default_index_url=None, index_urls=None, trusted_hosts=(), format_control=format_control) writer.write( results=results, reverse_dependencies=resolver.reverse_dependencies(results), unsafe_requirements=resolver.unsafe_constraints, primary_packages={key_from_ireq(ireq) for ireq in constraints}, markers={}, hashes=None) with open(tmp_file_name, 'r') as tmp_file: return tmp_file.readlines() finally: os.unlink(tmp_file_name)
def test_iter_lines__unsafe_dependencies(from_line, allow_unsafe): writer = OutputWriter( src_files=["src_file", "src_file2"], dst_file="dst_file", dry_run=True, emit_header=True, emit_index=True, emit_trusted_host=True, annotate=True, generate_hashes=False, default_index_url=None, index_urls=[], trusted_hosts=[], format_control=FormatControl(set(), set()), allow_unsafe=allow_unsafe, ) ireq = [from_line('test==1.2')] unsafe_req = [from_line('setuptools')] reverse_dependencies = {'test': ['xyz']} str_lines = list(writer._iter_lines( ireq, unsafe_req, reverse_dependencies, ['test'], {}, None, )) assert comment('# The following packages are considered to be unsafe in a requirements file:') in str_lines if allow_unsafe: assert 'setuptools' in str_lines else: assert comment('# setuptools') in str_lines assert 'test==1.2' in str_lines
def writer(tmpdir_cwd): with open("src_file", "w"), open("src_file2", "w"): pass cli_args = [ "--dry-run", "--output-file", "requirements.txt", "src_file", "src_file2", ] with cli.make_context("pip-compile", cli_args) as ctx: writer = OutputWriter( src_files=["src_file", "src_file2"], dst_file=ctx.params["output_file"], click_ctx=ctx, dry_run=True, emit_header=True, emit_index=True, emit_trusted_host=True, annotate=True, generate_hashes=False, default_index_url=None, index_urls=[], trusted_hosts=[], format_control=FormatControl(set(), set()), allow_unsafe=False, find_links=[], emit_find_links=True, ) yield writer
def writer(): return OutputWriter(src_files=["src_file", "src_file2"], dst_file="dst_file", dry_run=True, emit_header=True, emit_index=True, annotate=True, generate_hashes=False, default_index_url=None, index_urls=[], trusted_hosts=[], format_control=FormatControl(set(), set()))
def writer(): return OutputWriter(src_file="src_file", dry_run=True, emit_header=True, emit_index=True, annotate=True, default_index_url=None, index_urls=[])
def writer(tmp_file): return OutputWriter( src_files=["src_file", "src_file2"], dst_file=tmp_file, dry_run=True, emit_header=True, emit_index=True, emit_trusted_host=True, annotate=True, generate_hashes=False, default_index_url=None, index_urls=[], trusted_hosts=[], format_control=FormatControl(set(), set()), allow_unsafe=False, find_links=[], )