def unpack_tuple_to_one(editor=wingapi.kArgEditor):
'''
Turn `things` into `(thing,)`.
Useful for writing things like:
(thing,) == things
See this blog post for more context: http://blog.ram.rachum.com/post/1198230058/python-idiom-for-taking-the-single-item-from-a-list
Suggested key combination: `Insert U`
'''
assert isinstance(editor, wingapi.CAPIEditor)
document = editor.GetDocument()
assert isinstance(document, wingapi.CAPIDocument)
with shared.UndoableAction(document):
start, end = shared.select_current_word(editor)
plural_word = document.GetCharRange(start, end)
if not plural_word.endswith('s'):
return
singular_word = shared.plural_word_to_singular_word(plural_word)
segment_to_insert = '(%s,)' % singular_word
document.DeleteChars(start, end - 1)
document.InsertChars(start, segment_to_insert)
new_position = start + len(segment_to_insert)
editor.SetSelection(new_position, new_position)
def unpack_tuple_to_one(editor=wingapi.kArgEditor):
'''
Turn `things` into `(thing,)`.
Useful for writing things like:
(thing,) == things
See this blog post for more context: http://blog.ram.rachum.com/post/1198230058/python-idiom-for-taking-the-single-item-from-a-list
Suggested key combination: `Insert U`
'''
assert isinstance(editor, wingapi.CAPIEditor)
document = editor.GetDocument()
assert isinstance(document, wingapi.CAPIDocument)
with shared.UndoableAction(document):
start, end = shared.select_current_word(editor)
plural_word = document.GetCharRange(start, end)
if not plural_word.endswith('s'):
return
singular_word = shared.plural_word_to_singular_word(plural_word)
segment_to_insert = '(%s,)' % singular_word
document.DeleteChars(start, end - 1)
document.InsertChars(start, segment_to_insert)
new_position = start + len(segment_to_insert)
editor.SetSelection(new_position, new_position)
def for_thing_in_things(editor=wingapi.kArgEditor, comprehension=False):
'''
Turn `things` into `for thing in things:`.
Type any pluarl word, like `bananas` or `directories`. Then run this
script, and you get `for directory in directories`.
This also works for making `range(number)` into `for i in range(number):`.
Note: The `:` part is added only on Windows.
If `comprehension=True`, inputs `for thing in things` without the colon and
puts the caret before instead of after.
Suggested key combination: `Insert Ctrl-F`
`Insert Shift-F` for `comprehension=True`
'''
assert isinstance(editor, wingapi.CAPIEditor)
document = editor.GetDocument()
document_text = shared.get_text(document)
assert isinstance(document, wingapi.CAPIDocument)
with shared.UndoableAction(document):
end_position, _ = editor.GetSelection()
line_number = document.GetLineNumberFromPosition(end_position)
line_start = document.GetLineStart(line_number)
line_end = document.GetLineEnd(line_number)
line_contents = document.GetCharRange(line_start, line_end)
editor.SetSelection(end_position, end_position)
if ')' in document.GetCharRange(end_position - 1, end_position + 1) \
and 'range(' in line_contents:
start_position = document_text.find('range(', line_start)
end_position = document_text.find(
')',
start_position,
min((line_end, end_position + 1))
) + 1
else:
wingapi.gApplication.ExecuteCommand('backward-word')
start_position, _ = editor.GetSelection()
print(start_position, end_position)
base_text = document.GetCharRange(start_position, end_position)
print(base_text)
### Analyzing base text: ##############################################
# #
if range_pattern.match(base_text):
variable_name = 'i'
elif base_text.endswith('s'):
variable_name = shared.plural_word_to_singular_word(base_text)
# #
### Finished analyzing base text. #####################################
segment_to_insert = 'for %s in ' % variable_name
if comprehension:
segment_to_insert = ' %s' % segment_to_insert
document.InsertChars(start_position, segment_to_insert)
if comprehension:
editor.SetSelection(start_position, start_position)
else:
editor.ExecuteCommand('end-of-line')
if shared.autopy_available:
import autopy.key
autopy.key.tap(':')
def for_thing_in_things(editor=wingapi.kArgEditor, app=wingapi.kArgApplication,
comprehension=False):
'''
Turn `things` into `for thing in things:`.
Type any pluarl word, like `bananas` or `directories`. Then run this
script, and you get `for directory in directories`.
This also works for making `range(number)` into `for i in range(number):`.
Note: The `:` part is added only on Windows.
If `comprehension=True`, inputs `for thing in things` without the colon and
puts the caret before instead of after.
Suggested key combination: `Insert Ctrl-F`
`Insert Shift-F` for `comprehension=True`
'''
assert isinstance(editor, wingapi.CAPIEditor)
document = editor.GetDocument()
document_text = shared.get_text(document)
assert isinstance(document, wingapi.CAPIDocument)
with shared.UndoableAction(document):
end_position, _ = editor.GetSelection()
line_number = document.GetLineNumberFromPosition(end_position)
line_start = document.GetLineStart(line_number)
line_end = document.GetLineEnd(line_number)
line_contents = document.GetCharRange(line_start, line_end)
editor.SetSelection(end_position, end_position)
if ')' in document.GetCharRange(end_position - 1, end_position + 1) \
and 'range(' in line_contents:
text_start_position = document_text.find('range(', line_start)
if document_text[text_start_position - 1] == 'x':
text_start_position -= 1
end_position = document_text.find(
')',
text_start_position,
min((line_end, end_position + 1))
) + 1
else:
wingapi.gApplication.ExecuteCommand('backward-word')
text_start_position, _ = editor.GetSelection()
wingapi.gApplication.ExecuteCommand('forward-word')
wingapi.gApplication.ExecuteCommand('select-expression')
shared.strip_selection_if_single_line(editor)
expression_start_position, _ = editor.GetSelection()
base_text = document.GetCharRange(text_start_position, end_position)
### Analyzing base text: ##############################################
# #
if range_pattern.match(base_text):
variable_name = 'i'
elif base_text.endswith('s'):
variable_name = shared.plural_word_to_singular_word(base_text)
else:
raise Exception('This text doesn\'t work: %s' % base_text)
# #
### Finished analyzing base text. #####################################
segment_to_insert = 'for %s in ' % variable_name
with shared.SelectionRestorer(editor):
app.ExecuteCommand('beginning-of-line-text(toggle=False)')
home_position, _ = editor.GetSelection()
if comprehension:
segment_to_insert = ' %s' % segment_to_insert
document.InsertChars(expression_start_position, segment_to_insert)
editor.SetSelection(expression_start_position,
expression_start_position)
else:
document.InsertChars(home_position, segment_to_insert)
editor.ExecuteCommand('end-of-line')
if shared.autopy_available:
import autopy.key
autopy.key.tap(':')
