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
|
43
|
12 log = logging.getLogger(__name__)
|
|
13
|
|
14 class AdminController(BaseController):
|
|
15
|
|
16
|
|
17 def __before__(self):
|
|
18 c.staticurl = g.statics
|
44
|
19 c.admin_user = True
|
43
|
20
|
|
21 def index(self):
|
|
22 # Return a rendered template
|
|
23 return render('/admin.html')
|
|
24
|
44
|
25 def repos_manage(self):
|
|
26 return render('/repos_manage.html')
|
|
27
|
|
28 def users_manage(self):
|
|
29 conn, cur = auth.get_sqlite_conn_cur()
|
|
30 cur.execute('SELECT * FROM users')
|
|
31 c.users_list = cur.fetchall()
|
|
32 return render('/users_manage.html')
|
|
33
|
43
|
34 def manage_hgrc(self):
|
|
35 pass
|
|
36
|
|
37 def hgrc(self, dirname):
|
|
38 filename = os.path.join(dirname, '.hg', 'hgrc')
|
|
39 return filename
|
|
40
|
|
41 def add_repo(self, new_repo):
|
|
42
|
|
43
|
|
44 #extra check it can be add since it's the command
|
44
|
45 if new_repo == '_admin':
|
|
46 c.msg = 'DENIED'
|
43
|
47 c.new_repo = ''
|
|
48 return render('add.html')
|
|
49
|
|
50 new_repo = new_repo.replace(" ", "_")
|
|
51 new_repo = new_repo.replace("-", "_")
|
|
52
|
|
53 try:
|
|
54 self._create_repo(new_repo)
|
|
55 c.new_repo = new_repo
|
|
56 c.msg = 'added repo'
|
|
57 except Exception as e:
|
|
58 c.new_repo = 'Exception when adding: %s' % new_repo
|
|
59 c.msg = str(e)
|
|
60
|
|
61 return render('add.html')
|
|
62
|
|
63 def _check_repo(self, repo_name):
|
|
64 p = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
|
|
65 config_path = os.path.join(p, 'hgwebdir.config')
|
|
66
|
|
67 cp = ConfigParser()
|
|
68
|
|
69 cp.read(config_path)
|
|
70 repos_path = cp.get('paths', '/').replace("**", '')
|
|
71
|
|
72 if not repos_path:
|
|
73 raise Exception('Could not read config !')
|
|
74
|
|
75 self.repo_path = os.path.join(repos_path, repo_name)
|
|
76
|
|
77 try:
|
|
78 r = hg.repository(ui.ui(), self.repo_path)
|
|
79 hg.verify(r)
|
|
80 #here we hnow that repo exists it was verified
|
|
81 log.info('%s repo is already created', repo_name)
|
|
82 raise Exception('Repo exists')
|
|
83 except RepoError:
|
|
84 log.info('%s repo is free for creation', repo_name)
|
|
85 #it means that there is no valid repo there...
|
|
86 return True
|
|
87
|
|
88
|
|
89 def _create_repo(self, repo_name):
|
|
90 if repo_name in [None, '', 'add']:
|
|
91 raise Exception('undefined repo_name of repo')
|
|
92
|
|
93 if self._check_repo(repo_name):
|
|
94 log.info('creating repo %s in %s', repo_name, self.repo_path)
|
|
95 cmd = """mkdir %s && hg init %s""" \
|
|
96 % (self.repo_path, self.repo_path)
|
|
97 os.popen(cmd)
|