Viewing file:  iobject.py (4.09 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish, dis-
# tribute, sublicense, and/or sell copies of the Software, and to permit
# persons to whom the Software is furnished to do so, subject to the fol-
# lowing conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
import os
def int_val_fn(v):
try:
int(v)
return True
except:
return False
class IObject(object):
def choose_from_list(self, item_list, search_str='',
prompt='Enter Selection'):
if not item_list:
print('No Choices Available')
return
choice = None
while not choice:
n = 1
choices = []
for item in item_list:
if isinstance(item, basestring):
print('[%d] %s' % (n, item))
choices.append(item)
n += 1
else:
obj, id, desc = item
if desc:
if desc.find(search_str) >= 0:
print('[%d] %s - %s' % (n, id, desc))
choices.append(obj)
n += 1
else:
if id.find(search_str) >= 0:
print('[%d] %s' % (n, id))
choices.append(obj)
n += 1
if choices:
val = raw_input('%s[1-%d]: ' % (prompt, len(choices)))
if val.startswith('/'):
search_str = val[1:]
else:
try:
int_val = int(val)
if int_val == 0:
return None
choice = choices[int_val-1]
except ValueError:
print('%s is not a valid choice' % val)
except IndexError:
print('%s is not within the range[1-%d]' % (val,
len(choices)))
else:
print("No objects matched your pattern")
search_str = ''
return choice
def get_string(self, prompt, validation_fn=None):
okay = False
while not okay:
val = raw_input('%s: ' % prompt)
if validation_fn:
okay = validation_fn(val)
if not okay:
print('Invalid value: %s' % val)
else:
okay = True
return val
def get_filename(self, prompt):
okay = False
val = ''
while not okay:
val = raw_input('%s: %s' % (prompt, val))
val = os.path.expanduser(val)
if os.path.isfile(val):
okay = True
elif os.path.isdir(val):
path = val
val = self.choose_from_list(os.listdir(path))
if val:
val = os.path.join(path, val)
okay = True
else:
val = ''
else:
print('Invalid value: %s' % val)
val = ''
return val
def get_int(self, prompt):
s = self.get_string(prompt, int_val_fn)
return int(s)
|