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))
|
|
33 if auth.authfunc(None, c.form_result['username'], c.form_result['password']) and\
|
|
34 c.form_result['username'] == 'admin':
|
|
35 session['admin_user'] = True
|
|
36 session['admin_username'] = c.form_result['username']
|
|
37 session.save()
|
|
38 return redirect(url('admin_home'))
|
|
39 else:
|
|
40 raise formencode.Invalid('Login Error', None, None,
|
|
41 error_dict={'username':'invalid login',
|
|
42 'password':'invalid password'})
|
|
43
|
|
44 except formencode.Invalid, error:
|
|
45 c.form_result = error.value
|
|
46 c.form_errors = error.error_dict or {}
|
|
47 html = render('/admin.html')
|
|
48
|
|
49 return htmlfill.render(
|
|
50 html,
|
|
51 defaults=c.form_result,
|
|
52 encoding="UTF-8"
|
|
53 )
|
43
|
54 return render('/admin.html')
|
|
55
|
44
|
56 def repos_manage(self):
|
|
57 return render('/repos_manage.html')
|
|
58
|
|
59 def users_manage(self):
|
|
60 conn, cur = auth.get_sqlite_conn_cur()
|
|
61 cur.execute('SELECT * FROM users')
|
|
62 c.users_list = cur.fetchall()
|
|
63 return render('/users_manage.html')
|
|
64
|
43
|
65 def manage_hgrc(self):
|
|
66 pass
|
|
67
|
|
68 def hgrc(self, dirname):
|
|
69 filename = os.path.join(dirname, '.hg', 'hgrc')
|
|
70 return filename
|
|
71
|
|
72 def add_repo(self, new_repo):
|
|
73
|
|
74
|
|
75 #extra check it can be add since it's the command
|
44
|
76 if new_repo == '_admin':
|
|
77 c.msg = 'DENIED'
|
43
|
78 c.new_repo = ''
|
|
79 return render('add.html')
|
|
80
|
|
81 new_repo = new_repo.replace(" ", "_")
|
|
82 new_repo = new_repo.replace("-", "_")
|
|
83
|
|
84 try:
|
|
85 self._create_repo(new_repo)
|
|
86 c.new_repo = new_repo
|
|
87 c.msg = 'added repo'
|
|
88 except Exception as e:
|
|
89 c.new_repo = 'Exception when adding: %s' % new_repo
|
|
90 c.msg = str(e)
|
|
91
|
|
92 return render('add.html')
|
|
93
|
|
94 def _check_repo(self, repo_name):
|
|
95 p = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
|
|
96 config_path = os.path.join(p, 'hgwebdir.config')
|
|
97
|
|
98 cp = ConfigParser()
|
|
99
|
|
100 cp.read(config_path)
|
|
101 repos_path = cp.get('paths', '/').replace("**", '')
|
|
102
|
|
103 if not repos_path:
|
|
104 raise Exception('Could not read config !')
|
|
105
|
|
106 self.repo_path = os.path.join(repos_path, repo_name)
|
|
107
|
|
108 try:
|
|
109 r = hg.repository(ui.ui(), self.repo_path)
|
|
110 hg.verify(r)
|
|
111 #here we hnow that repo exists it was verified
|
|
112 log.info('%s repo is already created', repo_name)
|
|
113 raise Exception('Repo exists')
|
|
114 except RepoError:
|
|
115 log.info('%s repo is free for creation', repo_name)
|
|
116 #it means that there is no valid repo there...
|
|
117 return True
|
|
118
|
|
119
|
|
120 def _create_repo(self, repo_name):
|
|
121 if repo_name in [None, '', 'add']:
|
|
122 raise Exception('undefined repo_name of repo')
|
|
123
|
|
124 if self._check_repo(repo_name):
|
|
125 log.info('creating repo %s in %s', repo_name, self.repo_path)
|
|
126 cmd = """mkdir %s && hg init %s""" \
|
|
127 % (self.repo_path, self.repo_path)
|
|
128 os.popen(cmd)
|