sangge/waka-readme
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a try/except block for catching errors (#14)

Browse Source 
* Add a try/except block for catching errors

* Refactored the try/except to catch the GitHub error
This commit is contained in:
Athul Cyriac Ajay 2020-07-20 11:18:23 +05:30 committed by GitHub
parent 0edb69508c
commit 744bf5ae56
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
main.py
View File

@ -7,8 +7,7 @@ import os
import base64 import base64
import datetime import datetime
import requests import requests
from github import Github from github import Github, GithubException
START_COMMENT = '<!--START_SECTION:waka-->' START_COMMENT = '<!--START_SECTION:waka-->'
END_COMMENT = '<!--END_SECTION:waka-->' END_COMMENT = '<!--END_SECTION:waka-->'
@ -38,7 +37,10 @@ def get_stats() -> str:
'''Gets API data and returns markdown progress''' '''Gets API data and returns markdown progress'''
data = requests.get( data = requests.get(
f"https://wakatime.com/api/v1/users/current/stats/last_7_days?api_key={waka_key}").json() f"https://wakatime.com/api/v1/users/current/stats/last_7_days?api_key={waka_key}").json()
lang_data = data['data']['languages'] try:
lang_data = data['data']['languages']
except KeyError:
print("Please Add your Wakatime API Key to the Repository Secrets")
data_list = [] data_list = []
for l in lang_data[:5]: for l in lang_data[:5]:
ln = len(l['name']) ln = len(l['name'])
@ -70,11 +72,14 @@ def generate_new_readme(stats: str, readme: str) ->str:
if __name__ == '__main__': if __name__ == '__main__':
g = Github(ghtoken) g = Github(ghtoken)
repo = g.get_repo(f"{user}/{user}") try:
repo = g.get_repo(f"{user}/{user}")
except GithubException:
print("Authentication Error. Try saving a GitHub Token in your Repo Secrets or Use the GitHub Actions Token, which is automatically used by the action.")
contents = repo.get_readme() contents = repo.get_readme()
waka_stats = get_stats() waka_stats = get_stats()
rdmd = decode_readme(contents.content) rdmd = decode_readme(contents.content)
new_readme = generate_new_readme(stats=waka_stats, readme=rdmd) new_readme = generate_new_readme(stats=waka_stats, readme=rdmd)
if new_readme != rdmd: if new_readme != rdmd:
repo.update_file(path=contents.path, message='Updated with Dev Metrics', repo.update_file(path=contents.path, message='Updated with Dev Metrics', content=new_readme, sha=contents.sha, branch='master')
content=new_readme, sha=contents.sha, branch='master')