Mercurial > public > src > rhodecode
annotate pylons_app/controllers/admin.py @ 47:f6ac79182600
Added rest controllers for repos and users,
templating changes + css fixes
| author | Marcin Kuzminski <marcin@python-blog.com> |
|---|---|
| date | Wed, 07 Apr 2010 20:19:25 +0200 |
| parents | 9db7782727b3 |
| children | 25e516447a33 |
| rev | line source |
|---|---|
| 43 | 1 import logging |
| 2 | |
| 3 from pylons import request, response, session, tmpl_context as c, url, app_globals as g | |
| 4 from pylons.controllers.util import abort, redirect | |
| 5 | |
| 6 from pylons_app.lib.base import BaseController, render | |
| 7 import os | |
| 8 from mercurial import ui, hg | |
| 9 from mercurial.error import RepoError | |
| 10 from ConfigParser import ConfigParser | |
| 44 | 11 from pylons_app.lib import auth |
| 45 | 12 from pylons_app.model.forms import LoginForm |
| 13 import formencode | |
| 14 import formencode.htmlfill as htmlfill | |
| 43 | 15 log = logging.getLogger(__name__) |
| 16 | |
| 17 class AdminController(BaseController): | |
| 18 | |
| 19 | |
| 20 def __before__(self): | |
| 21 c.staticurl = g.statics | |
| 45 | 22 c.admin_user = session.get('admin_user') |
| 23 c.admin_username = session.get('admin_username') | |
| 43 | 24 |
| 25 def index(self): | |
| 26 # Return a rendered template | |
| 45 | 27 if request.POST: |
| 28 #import Login Form validator class | |
| 29 login_form = LoginForm() | |
| 30 | |
| 31 try: | |
| 32 c.form_result = login_form.to_python(dict(request.params)) | |
|
46
9db7782727b3
Static files for production fixed
Marcin Kuzminski <marcin@python-blog.com>
parents:
45
diff
changeset
|
33 if auth.admin_auth(c.form_result['username'], c.form_result['password']): |
| 45 | 34 session['admin_user'] = True |
| 35 session['admin_username'] = c.form_result['username'] | |
| 36 session.save() | |
| 37 return redirect(url('admin_home')) | |
| 38 else: | |
| 39 raise formencode.Invalid('Login Error', None, None, | |
| 40 error_dict={'username':'invalid login', | |
| 41 'password':'invalid password'}) | |
| 42 | |
| 43 except formencode.Invalid, error: | |
| 44 c.form_result = error.value | |
| 45 c.form_errors = error.error_dict or {} | |
| 46 html = render('/admin.html') | |
| 47 | |
| 48 return htmlfill.render( | |
| 49 html, | |
| 50 defaults=c.form_result, | |
| 51 encoding="UTF-8" | |
| 52 ) | |
| 43 | 53 return render('/admin.html') |
| 54 | |
| 55 def hgrc(self, dirname): | |
| 56 filename = os.path.join(dirname, '.hg', 'hgrc') | |
| 57 return filename | |
| 58 | |
| 59 def add_repo(self, new_repo): | |
| 60 | |
| 61 | |
| 62 #extra check it can be add since it's the command | |
| 44 | 63 if new_repo == '_admin': |
| 64 c.msg = 'DENIED' | |
| 43 | 65 c.new_repo = '' |
| 66 return render('add.html') | |
| 67 | |
| 68 new_repo = new_repo.replace(" ", "_") | |
| 69 new_repo = new_repo.replace("-", "_") | |
| 70 | |
| 71 try: | |
| 72 self._create_repo(new_repo) | |
| 73 c.new_repo = new_repo | |
| 74 c.msg = 'added repo' | |
| 75 except Exception as e: | |
| 76 c.new_repo = 'Exception when adding: %s' % new_repo | |
| 77 c.msg = str(e) | |
| 78 | |
| 79 return render('add.html') | |
| 80 | |
| 81 def _check_repo(self, repo_name): | |
| 82 p = os.path.dirname(os.path.dirname(os.path.dirname(__file__))) | |
| 83 config_path = os.path.join(p, 'hgwebdir.config') | |
| 84 | |
| 85 cp = ConfigParser() | |
| 86 | |
| 87 cp.read(config_path) | |
| 88 repos_path = cp.get('paths', '/').replace("**", '') | |
| 89 | |
| 90 if not repos_path: | |
| 91 raise Exception('Could not read config !') | |
| 92 | |
| 93 self.repo_path = os.path.join(repos_path, repo_name) | |
| 94 | |
| 95 try: | |
| 96 r = hg.repository(ui.ui(), self.repo_path) | |
| 97 hg.verify(r) | |
| 98 #here we hnow that repo exists it was verified | |
| 99 log.info('%s repo is already created', repo_name) | |
| 100 raise Exception('Repo exists') | |
| 101 except RepoError: | |
| 102 log.info('%s repo is free for creation', repo_name) | |
| 103 #it means that there is no valid repo there... | |
| 104 return True | |
| 105 | |
| 106 | |
| 107 def _create_repo(self, repo_name): | |
| 108 if repo_name in [None, '', 'add']: | |
| 109 raise Exception('undefined repo_name of repo') | |
| 110 | |
| 111 if self._check_repo(repo_name): | |
| 112 log.info('creating repo %s in %s', repo_name, self.repo_path) | |
| 113 cmd = """mkdir %s && hg init %s""" \ | |
| 114 % (self.repo_path, self.repo_path) | |
| 115 os.popen(cmd) |