Skip to content

Commit

Permalink
fix(package metadata): ensure package name is fixed
Browse files Browse the repository at this point in the history
Resolves #68
  • Loading branch information
mostaphaRoudsari committed Feb 13, 2021
1 parent 48c158f commit 725fa0a
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions pollination_dsl/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,8 @@ def get_docker_image_from_dependency(package, dependency, owner):


def _get_package_readme(package_name: str) -> str:
package_data = importlib_metadata.metadata(package_name.replace('-', '_'))
package_name = name_to_pollination(package_name).replace('-', '_')
package_data = importlib_metadata.metadata(package_name)
long_description = package_data.get_payload()
if not long_description.strip():
content = package_data.get('Description')
Expand All @@ -145,7 +146,8 @@ def _get_package_readme(package_name: str) -> str:

def _get_package_owner(package_name: str) -> str:
"""Author field is used for package owner."""
package_data = importlib_metadata.metadata(package_name.replace('-', '_'))
package_name = name_to_pollination(package_name).replace('-', '_')
package_data = importlib_metadata.metadata(package_name)
owner = package_data.get('Author')
assert owner, \
'You must set the author of the package in setup.py to Pollination account owner'
Expand Down Expand Up @@ -237,7 +239,8 @@ def _get_package_version(package_data: Dict) -> str:


def _get_package_data(package_name: str) -> Dict:

"""Get package data as a dictionary."""
package_name = name_to_pollination(package_name).replace('-', '_')
package_data = importlib_metadata.metadata(package_name)

data = {
Expand Down

0 comments on commit 725fa0a

Please sign in to comment.