Files
XSStrike/core/prompt.py
2018-11-15 15:41:01 +05:30

22 lines
550 B
Python

import os
import tempfile
from core.config import defaultEditor
def prompt(default=None):
editor = os.environ.get('EDITOR', defaultEditor)
with tempfile.NamedTemporaryFile(mode='r+') as tmpfile:
if default:
tmpfile.write(default)
tmpfile.flush()
child_pid = os.fork()
is_child = child_pid == 0
if is_child:
os.execvp(editor, [editor, tmpfile.name])
else:
os.waitpid(child_pid, 0)
tmpfile.seek(0)
return tmpfile.read().strip()