1
# Copyright (C) 2010 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Unicode-compatible command-line splitter for all platforms."""
22
_whitespace_match = re.compile(u'\s', re.UNICODE).match
25
class _PushbackSequence(object):
26
def __init__(self, orig):
27
self._iter = iter(orig)
28
self._pushback_buffer = []
31
if len(self._pushback_buffer) > 0:
32
return self._pushback_buffer.pop()
34
return self._iter.next()
36
def pushback(self, char):
37
self._pushback_buffer.append(char)
43
class _Whitespace(object):
44
def process(self, next_char, context):
45
if _whitespace_match(next_char):
46
if len(context.token) > 0:
50
elif next_char in context.allowed_quote_chars:
52
return _Quotes(next_char, self)
53
elif next_char == u'\\':
54
return _Backslash(self)
56
context.token.append(next_char)
60
class _Quotes(object):
61
def __init__(self, quote_char, exit_state):
62
self.quote_char = quote_char
63
self.exit_state = exit_state
65
def process(self, next_char, context):
66
if next_char == u'\\':
67
return _Backslash(self)
68
elif next_char == self.quote_char:
69
return self.exit_state
71
context.token.append(next_char)
75
class _Backslash(object):
76
# See http://msdn.microsoft.com/en-us/library/bb776391(VS.85).aspx
77
def __init__(self, exit_state):
78
self.exit_state = exit_state
81
def process(self, next_char, context):
82
if next_char == u'\\':
85
elif next_char in context.allowed_quote_chars:
86
# 2N backslashes followed by a quote are N backslashes
87
context.token.append(u'\\' * (self.count/2))
88
# 2N+1 backslashes follwed by a quote are N backslashes followed by
89
# the quote which should not be processed as the start or end of
91
if self.count % 2 == 1:
92
# odd number of \ escapes the quote
93
context.token.append(next_char)
95
# let exit_state handle next_char
96
context.seq.pushback(next_char)
98
return self.exit_state
100
# N backslashes not followed by a quote are just N backslashes
102
context.token.append(u'\\' * self.count)
104
# let exit_state handle next_char
105
context.seq.pushback(next_char)
106
return self.exit_state
108
def finish(self, context):
110
context.token.append(u'\\' * self.count)
114
def process(self, next_char, context):
115
if _whitespace_match(next_char):
117
elif next_char in context.allowed_quote_chars:
118
return _Quotes(next_char, self)
119
elif next_char == u'\\':
120
return _Backslash(self)
122
context.token.append(next_char)
126
class Splitter(object):
127
def __init__(self, command_line, single_quotes_allowed):
128
self.seq = _PushbackSequence(command_line)
129
self.allowed_quote_chars = u'"'
130
if single_quotes_allowed:
131
self.allowed_quote_chars += u"'"
137
quoted, token = self._get_token()
142
def _get_token(self):
145
state = _Whitespace()
146
for next_char in self.seq:
147
state = state.process(next_char, self)
150
if not state is None and not getattr(state, 'finish', None) is None:
152
result = u''.join(self.token)
153
if not self.quoted and result == '':
155
return self.quoted, result
158
def split(unsplit, single_quotes_allowed=True):
159
splitter = Splitter(unsplit, single_quotes_allowed=single_quotes_allowed)
160
return [arg for quoted, arg in splitter]
added
removed
bzrlib/cmdline.py
