Mercurial > public > src > rhodecode
annotate pylons_app/controllers/admin.py @ 46:9db7782727b3
Static files for production fixed
Error handler for debug on, added
admin auth function authenticates only admins
changed creation of db
| author | Marcin Kuzminski <marcin@python-blog.com> |
|---|---|
| date | Wed, 07 Apr 2010 19:39:31 +0200 |
| parents | a886f5eba757 |
| children | f6ac79182600 |
| 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 | |
| 44 | 55 def repos_manage(self): |
| 56 return render('/repos_manage.html') | |
| 57 | |
| 58 def users_manage(self): | |
| 59 conn, cur = auth.get_sqlite_conn_cur() | |
| 60 cur.execute('SELECT * FROM users') | |
| 61 c.users_list = cur.fetchall() | |
| 62 return render('/users_manage.html') | |
| 63 | |
| 43 | 64 def manage_hgrc(self): |
| 65 pass | |
| 66 | |
| 67 def hgrc(self, dirname): | |
| 68 filename = os.path.join(dirname, '.hg', 'hgrc') | |
| 69 return filename | |
| 70 | |
| 71 def add_repo(self, new_repo): | |
| 72 | |
| 73 | |
| 74 #extra check it can be add since it's the command | |
| 44 | 75 if new_repo == '_admin': |
| 76 c.msg = 'DENIED' | |
| 43 | 77 c.new_repo = '' |
| 78 return render('add.html') | |
| 79 | |
| 80 new_repo = new_repo.replace(" ", "_") | |
| 81 new_repo = new_repo.replace("-", "_") | |
| 82 | |
| 83 try: | |
| 84 self._create_repo(new_repo) | |
| 85 c.new_repo = new_repo | |
| 86 c.msg = 'added repo' | |
| 87 except Exception as e: | |
| 88 c.new_repo = 'Exception when adding: %s' % new_repo | |
| 89 c.msg = str(e) | |
| 90 | |
| 91 return render('add.html') | |
| 92 | |
| 93 def _check_repo(self, repo_name): | |
| 94 p = os.path.dirname(os.path.dirname(os.path.dirname(__file__))) | |
| 95 config_path = os.path.join(p, 'hgwebdir.config') | |
| 96 | |
| 97 cp = ConfigParser() | |
| 98 | |
| 99 cp.read(config_path) | |
| 100 repos_path = cp.get('paths', '/').replace("**", '') | |
| 101 | |
| 102 if not repos_path: | |
| 103 raise Exception('Could not read config !') | |
| 104 | |
| 105 self.repo_path = os.path.join(repos_path, repo_name) | |
| 106 | |
| 107 try: | |
| 108 r = hg.repository(ui.ui(), self.repo_path) | |
| 109 hg.verify(r) | |
| 110 #here we hnow that repo exists it was verified | |
| 111 log.info('%s repo is already created', repo_name) | |
| 112 raise Exception('Repo exists') | |
| 113 except RepoError: | |
| 114 log.info('%s repo is free for creation', repo_name) | |
| 115 #it means that there is no valid repo there... | |
| 116 return True | |
| 117 | |
| 118 | |
| 119 def _create_repo(self, repo_name): | |
| 120 if repo_name in [None, '', 'add']: | |
| 121 raise Exception('undefined repo_name of repo') | |
| 122 | |
| 123 if self._check_repo(repo_name): | |
| 124 log.info('creating repo %s in %s', repo_name, self.repo_path) | |
| 125 cmd = """mkdir %s && hg init %s""" \ | |
| 126 % (self.repo_path, self.repo_path) | |
| 127 os.popen(cmd) |