annotate pylons_app/controllers/hg.py @ 5:ad0dd3904225

added repo creation
author Marcin Kuzminski
date Thu, 18 Feb 2010 14:13:45 +0100
parents 564e40829f80
children 2620dac853ad
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
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
17
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):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
49 repos_path = '/home/marcink/python_workspace'
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
50 self.repo_path = os.path.join(repos_path, repo_name)
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
51
5
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
52 try:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
53 r = hg.repository(ui.ui(), self.repo_path)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
54 hg.verify(r)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
55 #here we hnow that repo exists it was verified
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
56 log.info('%s repo is already created', repo_name)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
57 raise Exception('Repo exists')
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
58 except RepoError:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
59 log.info('%s repo is free for creation', repo_name)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
60 #it means that there is no valid repo there...
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
61 return True
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
62
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
63
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
64 def _create_repo(self, repo_name):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
65 if repo_name in [None, '', 'add']:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
66 raise Exception('undefined repo_name of repo')
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
67
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
68 if self._check_repo(repo_name):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
69 log.info('creating repo %s', repo_name)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
70 cmd = """mkdir %s && hg init %s""" \
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
71 % (self.repo_path, self.repo_path)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
72 os.popen(cmd)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
73
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
74
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
75 def add_repo(self, new_repo):
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
76 tmpl = '''
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
77 <html>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
78 <body>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
79 %(msg)s%(new_repo)s!<br \>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
80 <a href="/">repos</a>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
81 </body>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
82 </html>
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
83 '''
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
84 #extra check it can be add since it's the command
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
85 if new_repo == 'add':
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
86 return [tmpl % ({'new_repo':'', 'msg':'you basstard ! this repo is a command'})]
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
87
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
88 new_repo = new_repo.replace(" ", "_")
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
89 new_repo = new_repo.replace("-", "_")
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
90
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
91 try:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
92 self._create_repo(new_repo)
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
93 except Exception as e:
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
94 return [tmpl % ({'new_repo':' Exception when adding: ' + new_repo, 'msg':str(e)})]
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
95
ad0dd3904225 added repo creation
Marcin Kuzminski
parents: 0
diff changeset
96 return [tmpl % ({'new_repo':new_repo, 'msg':'added repo: '})]
0
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
97
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
98 def view(self, environ, start_response):
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
99 #the following is only needed when using hgwebdir
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
100 app = _make_app()
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
101 #return wsgi_app(environ, start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
102 response = app(request.environ, self.start_response)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
103
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
104 if environ['PATH_INFO'].find("static") != -1:
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
105 return response
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
106 else:
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
107 #wrap the murcurial response in a mako template.
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
108 template = Template("".join(response),
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
109 lookup = environ['pylons.pylons']\
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
110 .config['pylons.g'].mako_lookup)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
111
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
112 return template.render(g = g, c = c, session = session, h = h)
564e40829f80 initial commit.
Marcin Kuzminski
parents:
diff changeset
113