3 # Patchwork command line client
4 # Copyright (C) 2008 Nate Case <ncase@xes-inc.com>
6 # This file is part of the Patchwork package.
8 # Patchwork is free software; you can redistribute it and/or modify
9 # it under the terms of the GNU General Public License as published by
10 # the Free Software Foundation; either version 2 of the License, or
11 # (at your option) any later version.
13 # Patchwork is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU General Public License for more details.
18 # You should have received a copy of the GNU General Public License
19 # along with Patchwork; if not, write to the Free Software
20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
32 # Default Patchwork remote XML-RPC server URL
33 # This script will check the PW_XMLRPC_URL environment variable
34 # for the URL to access. If that is unspecified, it will fallback to
35 # the hardcoded default value specified here.
36 DEFAULT_URL = "http://patchwork/xmlrpc/"
37 CONFIG_FILES = [os.path.expanduser('~/.pwclientrc')]
40 """Filter for selecting patches."""
42 # These fields refer to specific objects, so they are special
43 # because we have to resolve them to IDs before passing the
44 # filter to the server
48 # The dictionary that gets passed to via XML-RPC
51 def add(self, field, value):
54 elif field == 'project':
60 def resolve_ids(self, rpc):
61 """Resolve State, Project, and Person IDs based on filter strings."""
63 id = state_id_by_name(rpc, self.state)
65 sys.stderr.write("Note: No State found matching %s*, " \
66 "ignoring filter\n" % self.state)
68 self.d['state_id'] = id
70 if self.project != "":
71 id = project_id_by_name(rpc, self.project)
73 sys.stderr.write("Note: No Project found matching %s, " \
74 "ignoring filter\n" % self.project)
76 self.d['project_id'] = id
79 """Return human-readable description of the filter."""
82 class BasicHTTPAuthTransport(xmlrpclib.SafeTransport):
84 def __init__(self, username = None, password = None, use_https = False):
85 self.username = username
86 self.password = password
87 self.use_https = use_https
88 xmlrpclib.SafeTransport.__init__(self)
90 def authenticated(self):
91 return self.username != None and self.password != None
93 def send_host(self, connection, host):
94 xmlrpclib.Transport.send_host(self, connection, host)
95 if not self.authenticated():
97 credentials = '%s:%s' % (self.username, self.password)
98 auth = 'Basic ' + base64.encodestring(credentials).strip()
99 connection.putheader('Authorization', auth)
101 def make_connection(self, host):
103 fn = xmlrpclib.SafeTransport.make_connection
105 fn = xmlrpclib.Transport.make_connection
106 return fn(self, host)
109 sys.stderr.write("Usage: %s <action> [options]\n\n" % \
110 (os.path.basename(sys.argv[0])))
111 sys.stderr.write("Where <action> is one of:\n")
113 """ apply <ID> : Apply a patch (in the current dir, using -p1)
114 get <ID> : Download a patch and save it locally
115 projects : List all projects
116 states : Show list of potential patch states
117 list [str] : List patches, using the optional filters specified
118 below and an optional substring to search for patches
120 search [str] : Same as 'list'
121 view <ID> : View a patch
122 update [-s state] [-c commit-ref] <ID>
124 sys.stderr.write("""\nFilter options for 'list' and 'search':
125 -s <state> : Filter by patch state (e.g., 'New', 'Accepted', etc.)
126 -p <project> : Filter by project name (see 'projects' for list)
127 -w <who> : Filter by submitter (name, e-mail substring search)
128 -d <who> : Filter by delegate (name, e-mail substring search)
129 -n <max #> : Restrict number of results\n""")
130 sys.stderr.write("""\nActions that take an ID argument can also be \
132 -h <hash> : Lookup by patch hash\n""")
135 def project_id_by_name(rpc, linkname):
136 """Given a project short name, look up the Project ID."""
137 if len(linkname) == 0:
139 projects = rpc.project_list(linkname, 0)
140 for project in projects:
141 if project['linkname'] == linkname:
145 def state_id_by_name(rpc, name):
146 """Given a partial state name, look up the state ID."""
149 states = rpc.state_list(name, 0)
151 if state['name'].lower().startswith(name.lower()):
155 def person_ids_by_name(rpc, name):
156 """Given a partial name or email address, return a list of the
157 person IDs that match."""
160 people = rpc.person_list(name, 0)
161 return map(lambda x: x['id'], people)
163 def list_patches(patches):
164 """Dump a list of patches to stdout."""
165 print("%-5s %-12s %s" % ("ID", "State", "Name"))
166 print("%-5s %-12s %s" % ("--", "-----", "----"))
167 for patch in patches:
168 print("%-5d %-12s %s" % (patch['id'], patch['state'], patch['name']))
170 def action_list(rpc, filter, submitter_str, delegate_str):
171 filter.resolve_ids(rpc)
173 if submitter_str != "":
174 ids = person_ids_by_name(rpc, submitter_str)
176 sys.stderr.write("Note: Nobody found matching *%s*\n" % \
180 person = rpc.person_get(id)
181 print "Patches submitted by %s <%s>:" % \
182 (unicode(person['name']).encode("utf-8"), \
183 unicode(person['email']).encode("utf-8"))
185 f.add("submitter_id", id)
186 patches = rpc.patch_list(f.d)
187 list_patches(patches)
190 if delegate_str != "":
191 ids = person_ids_by_name(rpc, delegate_str)
193 sys.stderr.write("Note: Nobody found matching *%s*\n" % \
197 person = rpc.person_get(id)
198 print "Patches delegated to %s <%s>:" % \
199 (person['name'], person['email'])
201 f.add("delegate_id", id)
202 patches = rpc.patch_list(f.d)
203 list_patches(patches)
206 patches = rpc.patch_list(filter.d)
207 list_patches(patches)
209 def action_projects(rpc):
210 projects = rpc.project_list("", 0)
211 print("%-5s %-24s %s" % ("ID", "Name", "Description"))
212 print("%-5s %-24s %s" % ("--", "----", "-----------"))
213 for project in projects:
214 print("%-5d %-24s %s" % (project['id'], \
215 project['linkname'], \
218 def action_states(rpc):
219 states = rpc.state_list("", 0)
220 print("%-5s %s" % ("ID", "Name"))
221 print("%-5s %s" % ("--", "----"))
223 print("%-5d %s" % (state['id'], state['name']))
225 def action_get(rpc, patch_id):
226 patch = rpc.patch_get(patch_id)
227 s = rpc.patch_get_mbox(patch_id)
229 if patch == {} or len(s) == 0:
230 sys.stderr.write("Unable to get patch %d\n" % patch_id)
233 base_fname = fname = os.path.basename(patch['filename'])
235 while os.path.exists(fname):
236 fname = "%s.%d" % (base_fname, i)
242 sys.stderr.write("Unable to open %s for writing\n" % fname)
246 f.write(unicode(s).encode("utf-8"))
248 print "Saved patch to %s" % fname
250 sys.stderr.write("Failed to write to %s\n" % fname)
253 def action_apply(rpc, patch_id):
254 patch = rpc.patch_get(patch_id)
256 sys.stderr.write("Error getting information on patch ID %d\n" % \
259 print "Applying patch #%d to current directory" % patch_id
260 print "Description: %s" % patch['name']
261 s = rpc.patch_get_mbox(patch_id)
263 proc = subprocess.Popen(['patch', '-p1'], stdin = subprocess.PIPE)
266 sys.stderr.write("Error: No patch content found\n")
269 def action_update_patch(rpc, patch_id, state = None, commit = None):
270 patch = rpc.patch_get(patch_id)
272 sys.stderr.write("Error getting information on patch ID %d\n" % \
279 state_id = state_id_by_name(rpc, state)
281 sys.stderr.write("Error: No State found matching %s*\n" % state)
283 params['state'] = state_id
286 params['commit_ref'] = commit
290 success = rpc.patch_set(patch_id, params)
291 except xmlrpclib.Fault, f:
292 sys.stderr.write("Error updating patch: %s\n" % f.faultString)
295 sys.stderr.write("Patch not updated\n")
297 def patch_id_from_hash(rpc, project, hash):
299 patch = rpc.patch_get_by_project_hash(project, hash)
300 except xmlrpclib.Fault:
301 # the server may not have the newer patch_get_by_project_hash function,
302 # so fall back to hash-only.
303 patch = rpc.patch_get_by_hash(hash)
310 auth_actions = ['update']
314 opts, args = getopt.getopt(sys.argv[2:], 's:p:w:d:n:c:h:')
315 except getopt.GetoptError, err:
319 if len(sys.argv) < 2:
322 action = sys.argv[1].lower()
334 config = ConfigParser.ConfigParser()
335 config.read(CONFIG_FILES)
337 # grab settings from config files
338 if config.has_option('base', 'url'):
339 url = config.get('base', 'url')
341 if config.has_option('base', 'project'):
342 project_str = config.get('base', 'project')
344 for name, value in opts:
350 submitter_str = value
359 filt.add("max_count", int(value))
361 sys.stderr.write("Invalid maximum count '%s'\n" % value)
364 sys.stderr.write("Unknown option '%s'\n" % name)
368 sys.stderr.write("Too many arguments specified\n")
371 (username, password) = (None, None)
373 if action in auth_actions:
374 if config.has_option('auth', 'username') and \
375 config.has_option('auth', 'password'):
377 use_https = url.startswith('https')
379 transport = BasicHTTPAuthTransport( \
380 config.get('auth', 'username'),
381 config.get('auth', 'password'),
385 sys.stderr.write(("The %s action requires authentication, "
386 "but no username or password\nis configured\n") % action)
390 filt.add("project", project_str)
393 filt.add("state", state_str)
396 rpc = xmlrpclib.Server(url, transport = transport)
398 sys.stderr.write("Unable to connect to %s\n" % url)
403 patch_id = patch_id_from_hash(rpc, project_str, hash_str)
405 sys.stderr.write("No patch has the hash provided\n")
409 if action == 'list' or action == 'search':
411 filt.add("name__icontains", args[0])
412 action_list(rpc, filt, submitter_str, delegate_str)
414 elif action.startswith('project'):
417 elif action.startswith('state'):
420 elif action == 'view':
422 patch_id = patch_id or int(args[0])
424 sys.stderr.write("Invalid patch ID given\n")
427 s = rpc.patch_get_mbox(patch_id)
429 print unicode(s).encode("utf-8")
431 elif action == 'get' or action == 'save':
433 patch_id = patch_id or int(args[0])
435 sys.stderr.write("Invalid patch ID given\n")
438 action_get(rpc, patch_id)
440 elif action == 'apply':
442 patch_id = patch_id or int(args[0])
444 sys.stderr.write("Invalid patch ID given\n")
447 action_apply(rpc, patch_id)
449 elif action == 'update':
451 patch_id = patch_id or int(args[0])
453 sys.stderr.write("Invalid patch ID given\n")
456 action_update_patch(rpc, patch_id, state = state_str,
460 sys.stderr.write("Unknown action '%s'\n" % action)
463 if __name__ == "__main__":