summaryrefslogtreecommitdiffstats
path: root/plugins/snippets/snippets/exporter.py
blob: 8f8249e8620900662b93eec2c4479defd4f6f783 (plain)
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
117
118
119
120
121
122
#    Gedit snippets plugin
#    Copyright (C) 2005-2006  Jesse van den Kieboom <jesse@icecrew.nl>
#
#    This program is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 2 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

import os
import tempfile
import shutil

import xml.etree.ElementTree as et
from . import helper

try:
    import gettext
    gettext.bindtextdomain('gedit')
    gettext.textdomain('gedit')
    _ = gettext.gettext
except:
    _ = lambda s: s

class Exporter:
    def __init__(self, filename, snippets):
        self.filename = filename
        self.set_snippets(snippets)

    def set_snippets(self, snippets):
        self.snippets = {}

        for snippet in snippets:
            lang = snippet.language()

            if lang in self.snippets:
                self.snippets[lang].append(snippet)
            else:
                self.snippets[lang] = [snippet]

    def export_xml(self, dirname, language, snippets):
        # Create the root snippets node
        root = et.Element('snippets')

        # Create filename based on language
        if language:
            filename = os.path.join(dirname, language + '.xml')

            # Set the language attribute
            root.attrib['language'] = language
        else:
            filename = os.path.join(dirname, 'global.xml')

        # Add all snippets to the root node
        for snippet in snippets:
            root.append(snippet.to_xml())

        # Write xml
        helper.write_xml(root, filename, ('text', 'accelerator'))

    def export_archive(self, cmd):
        dirname = tempfile.mkdtemp()

        # Save current working directory and change to temporary directory
        curdir = os.getcwd()

        try:
            os.chdir(dirname)

            # Write snippet xml files
            for language, snippets in self.snippets.items():
                self.export_xml(dirname, language , snippets)

            # Archive files
            status = os.system('%s "%s" *.xml' % (cmd, self.filename))
        finally:
            os.chdir(curdir)

        if status != 0:
            return _('The archive “%s” could not be created' % self.filename)

        # Remove the temporary directory
        shutil.rmtree(dirname)

    def export_targz(self):
        self.export_archive('tar -c --gzip -f')

    def export_tarbz2(self):
        self.export_archive('tar -c --bzip2 -f')

    def export_tar(self):
        self.export_archive('tar -cf')

    def run(self):
        dirname = os.path.dirname(self.filename)
        if not os.path.exists(dirname):
            return _('Target directory “%s” does not exist') % dirname

        if not os.path.isdir(dirname):
            return _('Target directory “%s” is not a valid directory') % dirname

        (root, ext) = os.path.splitext(self.filename)

        actions = {'.tar.gz': self.export_targz,
               '.tar.bz2': self.export_tarbz2,
               '.tar': self.export_tar}

        for k, v in actions.items():
            if self.filename.endswith(k):
                return v()

        return self.export_targz()

# ex:ts=4:et: