Skip to content

Commit

Permalink
Pulled back
Browse files Browse the repository at this point in the history
  • Loading branch information
randhircs committed Jan 16, 2025
1 parent b729dc8 commit 87c139a
Showing 1 changed file with 8 additions and 11 deletions.
19 changes: 8 additions & 11 deletions python/lib/dependabot/python/file_updater/setup_file_sanitizer.rb
Original file line number Diff line number Diff line change
Expand Up @@ -3,16 +3,13 @@

require "dependabot/python/file_updater"
require "dependabot/python/file_parser/setup_file_parser"
require "sorbet-runtime"

module Dependabot
module Python
class FileUpdater
# Take a setup.py, parses it (carefully!) and then create a new, clean
# setup.py using only the information which will appear in the lockfile.
class SetupFileSanitizer
extend T::Sig

def initialize(setup_file:, setup_cfg:)
@setup_file = setup_file
@setup_cfg = setup_cfg
Expand Down Expand Up @@ -42,7 +39,6 @@ def include_pbr?
end

def install_requires_array
@install_requires_array = T.let(T.untyped, T.untyped)
@install_requires_array ||=
parsed_setup_file.dependencies.filter_map do |dep|
next unless dep.requirements.first[:groups]
Expand Down Expand Up @@ -81,17 +77,18 @@ def extras_require_hash
end

def parsed_setup_file
@parsed_setup_file ||= Python::FileParser::SetupFileParser.new(
dependency_files: [
setup_file&.dup&.tap { |f| f.name = "setup.py" },
setup_cfg&.dup&.tap { |f| f.name = "setup.cfg" }
].compact
).dependency_set
@parsed_setup_file ||=
Python::FileParser::SetupFileParser.new(
dependency_files: [
setup_file&.dup&.tap { |f| f.name = "setup.py" },
setup_cfg&.dup&.tap { |f| f.name = "setup.cfg" }
].compact
).dependency_set
end

def package_name
content = setup_file.content
match = T.must(content).match(/name\s*=\s*['"](?<package_name>[^'"]+)['"]/)
match = content.match(/name\s*=\s*['"](?<package_name>[^'"]+)['"]/)
match ? match[:package_name] : "default_package_name"
end
end
Expand Down

0 comments on commit 87c139a

Please sign in to comment.