Permalink
Cannot retrieve contributors at this time
Fetching contributors…
Cannot retrieve contributors at this time
| from tornado import web, gen | |
| from tornado.escape import url_escape | |
| from docker.errors import NotFound | |
| from jupyterhub.handlers.base import BaseHandler | |
| from IPython.html.utils import url_path_join | |
| from tornado.httputil import url_concat | |
| from . import __version__ | |
| from .github_agent import * | |
| from .metrica import MetricaIdsMixin | |
| from datetime import datetime | |
| @gen.coroutine | |
| def is_repository_changed(user): | |
| try: | |
| setup = yield user.spawner.docker( | |
| 'exec_create', | |
| container=user.spawner.container_id, | |
| cmd="bash -c 'cd $JPY_WORKDIR && \ | |
| (git fetch --unshallow > /dev/null 2>&1; true) && \ | |
| git diff --name-only'", | |
| ) | |
| out = yield user.spawner.docker( | |
| 'exec_start', | |
| exec_id=setup['Id'], | |
| ) | |
| except NotFound: | |
| return False | |
| if out: | |
| return True | |
| else: | |
| return False | |
| @gen.coroutine | |
| def commit_container(request, spawner, log): | |
| image_tag = datetime.now().strftime('%Y-%m-%d_%H-%M-%S') | |
| image_name = 'everware_image/' + spawner.escaped_name + '/' + spawner.escaped_repo_url + '_' + spawner.container_id | |
| host_with_protocol = request.protocol + '://' + request.host | |
| url_with_image = url_concat(host_with_protocol + '/hub/spawn', | |
| dict(repourl='docker:' + image_name + ':' + image_tag)) | |
| log.info('Will commit %s' % url_with_image) | |
| commit = yield spawner.docker( | |
| 'commit', | |
| container=spawner.container_id, | |
| repository=image_name, | |
| tag=image_tag, | |
| message='Commit from control panel', | |
| author=spawner.escaped_name | |
| ) | |
| output_data = dict() | |
| if commit: | |
| output_data['url_with_image'] = url_with_image | |
| else: | |
| output_data['message'] = 'Sorry, can not save container' | |
| return output_data | |
| class HomeHandler(BaseHandler): | |
| """Render the user's home page.""" | |
| @web.authenticated | |
| @gen.coroutine | |
| def get(self): | |
| user = self.get_current_user() | |
| repourl = self.get_argument('repourl', '') | |
| all_arguments = {param: self.get_argument(param) for param in self.request.arguments} | |
| do_fork = self.get_argument('do_fork', False) | |
| do_push = self.get_argument('do_push', False) | |
| do_commit_container = self.get_argument('do_commit_container', False) | |
| notify_message = self.get_argument('message', '') | |
| notify_url_to_image = self.get_argument('url_with_image', '') | |
| if repourl: | |
| self.redirect(url_concat( | |
| url_path_join(self.hub.server.base_url, 'spawn'), all_arguments | |
| )) | |
| return | |
| branch_name = commit_sha = None | |
| repo_url = '' | |
| fork_exists = False | |
| repository_changed = False | |
| if user.running: | |
| branch_name = user.spawner.branch_name | |
| commit_sha = user.spawner.commit_sha | |
| repo_url = user.spawner.repo_url | |
| if user.running and do_commit_container: | |
| output_data = yield commit_container(self.request, user.spawner, self.log) | |
| self.redirect(url_concat('/hub/home', output_data)) | |
| if user.running and getattr(user, 'login_service', '') == 'github': | |
| if do_fork: | |
| self.log.info('Will fork %s' % user.spawner.repo_url) | |
| result = yield fork_repo( | |
| user.spawner, | |
| user.token | |
| ) | |
| self.redirect(url_concat('/hub/home', dict(message='Successfully forked'))) | |
| return | |
| if do_push: | |
| self.log.info('Will push to fork') | |
| result = yield push_repo( | |
| user, | |
| user.spawner, | |
| user.token | |
| ) | |
| result = str(result, 'ascii') | |
| self.log.info('Got after push: %s' % result) | |
| message = 'Successfully pushed' | |
| if 'Update through everware' not in result: | |
| message = result | |
| self.redirect(url_concat('/hub/home', dict(message='Push result: %s' % message))) | |
| return | |
| fork_exists = yield does_fork_exist( | |
| user.name, | |
| user.spawner, | |
| user.token | |
| ) | |
| repository_changed = yield is_repository_changed(user) | |
| if hasattr(user, 'login_service'): | |
| loginservice = user.login_service | |
| else: | |
| loginservice = 'none' | |
| metrica = MetricaIdsMixin() | |
| g_id = metrica.g_analitics_id | |
| ya_id = metrica.ya_metrica_id | |
| html = self.render_template('home.html', | |
| user=user, | |
| repourl=repo_url, | |
| login_service=loginservice, | |
| fork_exists=fork_exists, | |
| repository_changed=repository_changed, | |
| branch_name=branch_name, | |
| commit_sha=commit_sha, | |
| notify_message=notify_message, | |
| notify_url_to_image=notify_url_to_image, | |
| version=__version__, | |
| g_analitics_id=g_id, | |
| ya_metrica_id=ya_id | |
| ) | |
| self.finish(html) | |