-
-
Notifications
You must be signed in to change notification settings - Fork 2
/
__init__.py
107 lines (87 loc) · 4.35 KB
/
__init__.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
#!/usr/bin/env python3
# Copyright (C) 2016 - 2021 Sylvia van Os <[email protected]>
#
# Pext emoji module 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 3 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, see <http://www.gnu.org/licenses/>.
import os.path
import re
import string
from xml.etree.ElementTree import parse
from pext_base import ModuleBase
from pext_helpers import Action
class Module(ModuleBase):
def init(self, settings, q):
self.settings = settings
self.q = q
self.entries = {}
self.display_entries = []
self.entry_info = {}
self.emoji_version_regex = re.compile('^E(\d+\.\d+)$')
self._get_entries()
def _get_entries(self):
self.emoji_translations = {}
translation_files = ['annotations-{}.xml'.format(self.settings['_locale']),
'annotationsDerived-{}.xml'.format(self.settings['_locale'])]
for translation_file in translation_files:
try:
emoji_translation_xml = parse(os.path.join(os.path.dirname(os.path.abspath(__file__)), translation_file)).getroot()[1]
for annotation in emoji_translation_xml:
if 'type' in annotation.attrib and annotation.attrib['type'] == 'tts':
self.emoji_translations[annotation.attrib['cp']] = annotation.text
except Exception:
pass
current_group = ""
current_subgroup = ""
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'emoji-test.txt'), 'r', encoding='utf-8') as f:
for line in f:
line = line.strip()
if not line:
continue
if line.startswith('# group: '):
current_group = string.capwords(line[len('# group :'):].strip())
current_subgroup = ""
elif line.startswith('# subgroup: '):
current_subgroup = string.capwords(line[len('# subgroup :'):].replace('-', ' ').strip())
elif not line.startswith('#'):
try:
if line.split('#')[0].split(';')[1].strip() != 'fully-qualified':
continue
emoji, version, code = line.split('#', 1)[1].strip().split(' ', 2)
try:
version = re.match(self.emoji_version_regex, version).groups()[0]
except IndexError:
version = "Unknown"
code = "{} {}".format(version, code)
try:
code = self.emoji_translations[emoji]
except KeyError:
pass
except IndexError:
continue
identifier = '{} {} ({} - {})'.format(emoji, string.capwords(code.strip()), current_group, current_subgroup)
self.entries[identifier] = emoji
self.entry_info[identifier] = "<h1>{}</h1><h2>{}</h2></p><b>Group:</b> {}<br/><b>Subgroup:</b> {}</p><b>Emoji version:</b> {}".format(emoji, string.capwords(code.strip()), current_group, current_subgroup, version)
self.display_entries = sorted(list(self.entries.keys()))
self.q.put([Action.replace_entry_list, self.display_entries])
self.q.put([Action.replace_entry_info_dict, self.entry_info])
def stop(self):
pass
def selection_made(self, selection):
if len(selection) == 0:
self.q.put([Action.replace_entry_list, self.display_entries])
self.q.put([Action.replace_entry_info_dict, self.entry_info])
elif len(selection) == 1:
self.q.put([Action.copy_to_clipboard, self.entries[selection[0]["value"]]])
self.q.put([Action.close])
def process_response(self, response):
pass