-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathImagePaste.py
116 lines (102 loc) · 3.31 KB
/
ImagePaste.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import sublime
import sublime_plugin
import os
import sys
import re
package_file = os.path.normpath(os.path.abspath(__file__))
package_path = os.path.dirname(package_file)
lib_path = os.path.join(package_path, "lib")
if lib_path not in sys.path:
sys.path.append(lib_path)
print(sys.path)
from PIL import ImageGrab
from PIL import ImageFile
from PIL import Image
class ImagePasteCommand(sublime_plugin.TextCommand):
def run(self, edit):
view = self.view
rel_fn = self.paste()
if not rel_fn:
view.run_command("paste")
return
for pos in view.sel():
# print("scope name: %r" % (view.scope_name(pos.begin())))
if 'text.html.markdown' in view.scope_name(pos.begin()):
view.insert(edit, pos.begin(), "" % rel_fn)
else:
view.insert(edit, pos.begin(), "%s" % rel_fn)
# only the first cursor add the path
break
def paste(self):
ImageFile.LOAD_TRUNCATED_IMAGES = True
im = ImageGrab.grabclipboard()
if im:
abs_fn, rel_fn = self.get_filename()
im.save(abs_fn,'PNG')
return rel_fn
else:
print('clipboard buffer is not image!')
return None
def get_filename(self):
view = self.view
filename = view.file_name()
# create dir in current path with the name of current filename
dirname, _ = os.path.splitext(filename)
# create new image file under currentdir/filename_without_ext/filename_without_ext%d.png
fn_without_ext = os.path.basename(dirname)
if not os.path.lexists(dirname):
os.mkdir(dirname)
i = 0
while True:
# relative file path
rel_filename = os.path.join("%s/%s%d.png" % (fn_without_ext, fn_without_ext, i))
# absolute file path
abs_filename = os.path.join(dirname, "%s%d.png" % ( fn_without_ext, i))
if not os.path.exists(abs_filename):
break
i += 1
print("save file: " + abs_filename + "\nrel " + rel_filename)
return abs_filename, rel_filename
class ImagePreviewCommand(sublime_plugin.TextCommand):
def __init__(self, *args):
# self.view = view
super(ImagePreviewCommand, self).__init__(*args)
# self.phantom_set = sublime.PhantomSet(self.view)
self.displayed = False
def get_line(self):
v = self.view
rows, _ = v.rowcol(v.size())
for row in range(rows+1):
pt = v.text_point(row, 0)
tp_line = v.line(pt)
line = v.substr(tp_line)
yield tp_line, line
raise StopIteration
def run(self, edit):
print("run phantom")
view = self.view
dirname = os.path.dirname(view.file_name())
for tp, line in self.get_line():
m=re.search(r'!\[([^\]]*)\]\(([^)]*)\)', line)
if m:
name, file1 = m.group(1), m.group(2)
message = ""
file2 = os.path.join(dirname, file1)
# print("%s = %s" % (name, file1))
region = tp
with Image.open(file2) as im:
# print("file: %s with size: %d %d" % (file1, im.width, im.height))
message = '''<body>
<img width="%d" height="%d" src="file://%s"></img>
</body>''' % (im.width, im.height, file2)
if len(name) == 0:
name = file1
# phantom = sublime.Phantom(region, messag e, sublime.LAYOUT_BLOCK)
print("message %s" % message)
if not self.displayed:
self.view.add_phantom(name, region, message, sublime.LAYOUT_BLOCK)
else:
self.view.erase_phantoms(name)
# self.phantom_set.update([phantom])
# view.show_popup('<img src="file://c://msys64/home/chenyu/diary/diary/diary8.jpg">')
self.displayed = not self.displayed