annotate pylons_app/controllers/hg.py @ 6:2620dac853ad

added autoconfig loading
author Marcin Kuzminski
date Thu, 18 Feb 2010 14:51:26 +0100
parents ad0dd3904225
children 3092016c6d0c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
1 #!/usr/bin/python
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
2 # -*- coding: utf-8 -*-
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
3 import logging
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
4 from pylons_app.lib.base import BaseController
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
5 from pylons import c, g, session, h, request
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
6 from mako.template import Template
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
7 from pprint import pprint
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
8 import os
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
9 #uncomment the following if you want to serve a single repo
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
10 #from mercurial.hgweb.hgweb_mod import hgweb
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
11 from mercurial.hgweb.hgwebdir_mod import hgwebdir
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
12 from mercurial.hgweb.request import wsgiapplication
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
13 log = logging.getLogger(__name__)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
14
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
15 from mercurial import ui, hg
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
16 from mercurial.error import RepoError
6
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
17 from ConfigParser import ConfigParser
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
18 #http://bel-epa.com/hg/
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
19 #def make_web_app():
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
20 # repos = "hgwebdir.config"
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
21 # hgwebapp = hgwebdir(repos)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
22 # return hgwebapp
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
23 #
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
24 #class HgController(BaseController):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
25 #
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
26 # def index(self):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
27 # hgapp = wsgiapplication(make_web_app)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
28 # return hgapp(request.environ, self.start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
29 #
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
30 # def view(self, *args, **kwargs):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
31 # return u'dupa'
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
32 # #pprint(request.environ)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
33 # hgapp = wsgiapplication(make_web_app)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
34 # return hgapp(request.environ, self.start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
35
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
36 def _make_app():
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
37 #for single a repo
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
38 #return hgweb("/path/to/repo", "Name")
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
39 repos = "hgwebdir.config"
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
40 return hgwebdir(repos)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
41
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
42 def wsgi_app(environ, start_response):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
43 start_response('200 OK', [('Content-type', 'text/html')])
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
44 return ['<html>\n<body>\nHello World!\n</body>\n</html>']
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
45
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
46 class HgController(BaseController):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
47
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
48 def _check_repo(self, repo_name):
6
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
49
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
50 p = os.path.dirname(__file__)
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
51 config_path = os.path.join(p, '../..', 'hgwebdir.config')
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
52 print config_path
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
53
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
54 cp = ConfigParser()
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
55
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
56 cp.read(config_path)
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
57 repos_path = cp.get('paths', '/').replace("**", '')
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
58
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
59 if not repos_path:
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
60 raise Exception('Could not read config !')
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
61
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
62 self.repo_path = os.path.join(repos_path, repo_name)
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
63
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
64 try:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
65 r = hg.repository(ui.ui(), self.repo_path)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
66 hg.verify(r)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
67 #here we hnow that repo exists it was verified
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
68 log.info('%s repo is already created', repo_name)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
69 raise Exception('Repo exists')
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
70 except RepoError:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
71 log.info('%s repo is free for creation', repo_name)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
72 #it means that there is no valid repo there...
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
73 return True
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
74
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
75
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
76 def _create_repo(self, repo_name):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
77 if repo_name in [None, '', 'add']:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
78 raise Exception('undefined repo_name of repo')
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
79
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
80 if self._check_repo(repo_name):
6
2620dac853ad added autoconfig loading
Marcin Kuzminski
parents: 5
diff changeset
81 log.info('creating repo %s in %s', repo_name, self.repo_path)
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
82 cmd = """mkdir %s && hg init %s""" \
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
83 % (self.repo_path, self.repo_path)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
84 os.popen(cmd)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
85
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
86
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
87 def add_repo(self, new_repo):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
88 tmpl = '''
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
89 <html>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
90 <body>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
91 %(msg)s%(new_repo)s!<br \>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
92 <a href="/">repos</a>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
93 </body>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
94 </html>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
95 '''
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
96 #extra check it can be add since it's the command
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
97 if new_repo == 'add':
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
98 return [tmpl % ({'new_repo':'', 'msg':'you basstard ! this repo is a command'})]
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
99
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
100 new_repo = new_repo.replace(" ", "_")
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
101 new_repo = new_repo.replace("-", "_")
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
102
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
103 try:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
104 self._create_repo(new_repo)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
105 except Exception as e:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
106 return [tmpl % ({'new_repo':' Exception when adding: ' + new_repo, 'msg':str(e)})]
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
107
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
108 return [tmpl % ({'new_repo':new_repo, 'msg':'added repo: '})]
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
109
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
110 def view(self, environ, start_response):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
111 #the following is only needed when using hgwebdir
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
112 app = _make_app()
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
113 #return wsgi_app(environ, start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
114 response = app(request.environ, self.start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
115
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
116 if environ['PATH_INFO'].find("static") != -1:
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
117 return response
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
118 else:
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
119 #wrap the murcurial response in a mako template.
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
120 template = Template("".join(response),
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
121 lookup = environ['pylons.pylons']\
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
122 .config['pylons.g'].mako_lookup)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
123
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
124 return template.render(g = g, c = c, session = session, h = h)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
125