Commitd46ae36

committed

Add decorator for new methods requiring app creds

1 parent13b1d46 commitd46ae36Copy full SHA for d46ae36

File tree

+22

-0

lines changed

+22

-0

lines changed

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,28 @@ def auth_wrapper(self, args, *kwargs):`
`67`	`67`	`returnauth_wrapper`
`68`	`68`
`69`	`69`
	`70`	`+defrequires_app_credentials(func):`
	`71`	`+"""Require client_id and client_secret to be associated.`
	`72`	`+`
	`73`	`+ This is used to note and enforce which methods require a client_id and`
	`74`	`+ client_secret to be used.`
	`75`	`+`
	`76`	`+ """`
	`77`	`+@wraps(func)`
	`78`	`+defauth_wrapper(self,args,*kwargs):`
	`79`	`+ifhasattr(self,'_session')andself._session.params:`
	`80`	`+returnfunc(self,args,*kwargs)`
	`81`	`+else:`
	`82`	`+fromgithub3.modelsimportGitHubError`
	`83`	`+# Mock a 401 response`
	`84`	`+r=generate_fake_error_response(`
	`85`	`+'{"message": "Requires username/password authentication"}'`
	`86`	`+ )`
	`87`	`+raiseGitHubError(r)`
	`88`	`+`
	`89`	`+returnauth_wrapper`
	`90`	`+`
	`91`	`+`
`70`	`92`	`defgenerate_fake_error_response(msg,status_code=401,encoding='utf-8'):`
`71`	`93`	`r=Response()`
`72`	`94`	`r.status_code=status_code`

Comments

(0)