45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
import frappe
|
||
|
|
from frappe.data_migration.doctype.data_migration_connector.connectors.base import BaseConnection
|
||
|
|
from github import Github
|
||
|
|
|
||
|
|
class GithubConnection(BaseConnection):
|
||
|
|
def __init__(self, connector):
|
||
|
|
self.connector = connector
|
||
|
|
|
||
|
|
try:
|
||
|
|
password = self.get_password()
|
||
|
|
except frappe.AuthenticationError:
|
||
|
|
password = None
|
||
|
|
|
||
|
|
if self.connector.username and password:
|
||
|
|
self.connection = Github(self.connector.username, self.get_password())
|
||
|
|
else:
|
||
|
|
self.connection = Github()
|
||
|
|
|
||
|
|
self.name_field = 'id'
|
||
|
|
|
||
|
|
def insert(self, doctype, doc):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def update(self, doctype, doc, migration_id):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def delete(self, doctype, migration_id):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get(self, remote_objectname, fields=None, filters=None, start=0, page_length=10):
|
||
|
|
repo = filters.get('repo')
|
||
|
|
|
||
|
|
if remote_objectname == 'Milestone':
|
||
|
|
return self.get_milestones(repo, start, page_length)
|
||
|
|
if remote_objectname == 'Issue':
|
||
|
|
return self.get_issues(repo, start, page_length)
|
||
|
|
|
||
|
|
def get_milestones(self, repo, start=0, page_length=10):
|
||
|
|
_repo = self.connection.get_repo(repo)
|
||
|
|
return list(_repo.get_milestones()[start:start+page_length])
|
||
|
|
|
||
|
|
def get_issues(self, repo, start=0, page_length=10):
|
||
|
|
_repo = self.connection.get_repo(repo)
|
||
|
|
return list(_repo.get_issues()[start:start+page_length])
|