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
|
import os
import tempfile
import sys
import shutil
from .Library import *
class Importer:
def __init__(self, filename):
self.filename = filename
def import_destination(self, filename):
userdir = Library().userdir
filename = os.path.basename(filename)
(root, ext) = os.path.splitext(filename)
filename = os.path.join(userdir, root + ext)
i = 1
while os.path.exists(filename):
filename = os.path.join(userdir, root + '_' + str(i) + ext)
i += 1
return filename
def import_file(self, filename):
if not os.path.exists(filename):
return _('File "%s" does not exist') % filename
if not os.path.isfile(filename):
return _('File "%s" is not a valid snippets file') % filename
# Find destination for file to copy to
dest = self.import_destination(filename)
# Copy file
shutil.copy(filename, dest)
# Add library
if not Library().add_user_library(dest):
return _('Imported file "%s" is not a valid snippets file') % os.path.basename(dest)
def import_xml(self):
return self.import_file(self.filename)
def import_archive(self, cmd):
dirname = tempfile.mkdtemp()
status = os.system('cd %s; %s "%s"' % (dirname, cmd, self.filename))
if status != 0:
return _('The archive "%s" could not be extracted' % self.filename)
errors = []
# Now import all the files from the archive
for f in os.listdir(dirname):
f = os.path.join(dirname, f)
if os.path.isfile(f):
if self.import_file(f):
errors.append(os.path.basename(f))
else:
sys.stderr.write('Skipping %s, not a valid snippets file' % os.path.basename(f))
# Remove the temporary directory
shutil.rmtree(dirname)
if len(errors) > 0:
return _('The following files could not be imported: %s') % ', '.join(errors)
def import_targz(self):
self.import_archive('tar -x --gzip -f')
def import_tarbz2(self):
self.import_archive('tar -x --bzip2 -f')
def import_tar(self):
self.import_archive('tar -xf')
def run(self):
if not os.path.exists(self.filename):
return _('File "%s" does not exist') % self.filename
if not os.path.isfile(self.filename):
return _('File "%s" is not a valid snippets archive') % self.filename
(root, ext) = os.path.splitext(self.filename)
actions = {'.tar.gz': self.import_targz,
'.tar.bz2': self.import_tarbz2,
'.xml': self.import_xml,
'.tar': self.import_tar}
for k, v in actions.items():
if self.filename.endswith(k):
return v()
return _('File "%s" is not a valid snippets archive') % self.filename
# ex:ts=4:et:
|
