@@ -13,18 +13,22 @@ def clean_html_tags(text):
1313 clean_text = re .sub (tags , '' , text )
1414 return clean_text
1515
16- def generate_readme (metadata , folder , token = None , repo = None ):
16+ def generate_readme (metadata , folder , token = None , key = None , repo = None ):
1717 """
1818 Generate README.txt file.
1919
20- This function uses the template assets/README .txt
20+ This function uses the file README_template .txt in the repository root.
2121
2222 :param metadata: Project metadata from `get_metadata()`
2323 :type metadata: list
2424 :param folder: Path to QDR Prepared folder for project
2525 :type folder: String
2626 :param token: Dataverse token (required if the project is unpublished)
2727 :type token: string
28+ :param key: Github API key, or None for public repository
29+ :type key: String or None
30+ :param repo: Repository to use for the README template, defaults to dvcurator.hosts.curation_repo
31+ :type repo: String or None
2832
2933 :return: Path to newly generated README file
3034 :rtype: string
@@ -78,14 +82,22 @@ def generate_readme(metadata, folder, token=None, repo=None):
7882 'files' : dvcurator .fs .recursive_scan (folder ) #"\n".join(os.listdir(folder))
7983 }
8084
81- # the location of the template differs if this is a compiled pyinstaller file or run directly
82-
85+
8386 ## Download readme template from github
87+
88+
8489 host = "https://raw.githubusercontent.com/"
8590 repo = dvcurator .hosts .curation_repo if not repo else repo
8691 readme_url = host + repo + "/refs/heads/master/README_template.txt"
8792
88- response = requests .get (readme_url )
93+ if (not key ):
94+ print ("No github token set -- this will fail on private repo" )
95+ response = requests .get (readme_url )
96+ else :
97+ print ("Trying to download README template from private repository..." )
98+ key = {'Authorization' : "token " + key .strip ()}
99+ response = requests .get (readme_url , headers = key )
100+
89101 response .raise_for_status ()
90102 readme_template = response .text
91103 print ("Downloaded README template..." )
0 commit comments