111 lines
2.2 KiB
Python
Executable file
111 lines
2.2 KiB
Python
Executable file
#!/usr/bin/python3
|
|
|
|
import sys
|
|
import os
|
|
import random
|
|
import string
|
|
import argparse
|
|
|
|
def make_dir(d):
|
|
if not os.path.exists(d):
|
|
os.mkdir(d)
|
|
|
|
def delete_all(d):
|
|
files = list_all(d)
|
|
if not files:
|
|
return
|
|
x = input("Delete the above files? y/N ")
|
|
if x.lower() != 'y':
|
|
return
|
|
for file in files:
|
|
f = os.path.join(d,file)
|
|
os.remove(f)
|
|
print(f"removing: {f}")
|
|
print("Done")
|
|
|
|
def list_all(d):
|
|
if not os.path.exists(d):
|
|
print("Skratch path doesn't exist")
|
|
return 0
|
|
files = os.listdir(d)
|
|
if files == []:
|
|
print("No skratch files exist")
|
|
else:
|
|
for file in files:
|
|
print(file)
|
|
return files
|
|
|
|
def get_editor(use_v):
|
|
v = os.getenv("VISUAL")
|
|
e = os.getenv("EDITOR")
|
|
if (not v and not e):
|
|
e = "nano"
|
|
if use_v and v:
|
|
return v
|
|
return e
|
|
|
|
def mkstemp(fp):
|
|
if fp[-6:] != "XXXXXX":
|
|
print(f'Error: mkstemp got filepath: {fp}')
|
|
return None
|
|
for x in range(0,1000):
|
|
r = ''.join(random.choices(string.ascii_letters, k=6))
|
|
fp = fp[:-6] + r
|
|
if not os.path.exists(fp):
|
|
return fp
|
|
print("Error: Could not create new file, try running skratch -c and try again")
|
|
return None
|
|
|
|
|
|
def run(editor, sk_path, filename=None):
|
|
if not filename:
|
|
filename = "skratch-XXXXXX"
|
|
fp = os.path.join(sk_path, filename)
|
|
fp = mkstemp(fp)
|
|
else:
|
|
fp = os.path.join(sk_path, filename)
|
|
if fp != None:
|
|
os.execvp(editor, [editor,fp])
|
|
|
|
def main():
|
|
home_path = os.getenv("HOME")
|
|
sk_path = os.path.join(home_path, ".skratch")
|
|
make_dir(sk_path)
|
|
|
|
parser = argparse.ArgumentParser(
|
|
prog='skratch',
|
|
description='Creates a temp file and opens it in an editor',
|
|
epilog='Text at the bottom of help')
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument(
|
|
"-c",
|
|
help="delete all files",
|
|
action="store_true")
|
|
|
|
group.add_argument(
|
|
"-l",
|
|
help="list files",
|
|
action="store_true")
|
|
|
|
group.add_argument(
|
|
"-o",
|
|
help="open file",
|
|
action="store")
|
|
|
|
parser.add_argument(
|
|
"-v",
|
|
help="visual editor",
|
|
action="store_true")
|
|
|
|
args = parser.parse_args()
|
|
|
|
filename = "skratch-XXXXXX"
|
|
|
|
editor = get_editor(args.v)
|
|
if args.o: run(editor,sk_path,args.o)
|
|
elif args.c: delete_all(sk_path)
|
|
elif args.l: list_all(sk_path)
|
|
else: run(editor,sk_path)
|
|
|
|
main()
|
|
|