139 lines
4.1 KiB
Python
Executable File
139 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# Copyright 2017 Google Inc. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import argparse
|
|
import glob
|
|
import os
|
|
from os import path
|
|
import sys
|
|
|
|
"""Create aliases in target directory.
|
|
|
|
The target files should not contain the emoji variation selector
|
|
codepoint in their names."""
|
|
|
|
DATA_ROOT = path.dirname(path.abspath(__file__))
|
|
|
|
def str_to_seq(seq_str):
|
|
return tuple([int(s, 16) for s in seq_str.split('_')])
|
|
|
|
|
|
def seq_to_str(seq):
|
|
return '_'.join('%04x' % cp for cp in seq)
|
|
|
|
|
|
def read_emoji_aliases():
|
|
alias_path = path.join(DATA_ROOT, 'emoji_aliases.txt')
|
|
return read_emoji_aliases(alias_path)
|
|
|
|
|
|
def read_emoji_aliases(filename):
|
|
result = {}
|
|
|
|
with open(filename, 'r') as f:
|
|
for line in f:
|
|
ix = line.find('#')
|
|
if (ix > -1):
|
|
line = line[:ix]
|
|
line = line.strip()
|
|
if not line:
|
|
continue
|
|
als, trg = (s.strip() for s in line.split(';'))
|
|
try:
|
|
als_seq = tuple([int(x, 16) for x in als.split('_')])
|
|
trg_seq = tuple([int(x, 16) for x in trg.split('_')])
|
|
except:
|
|
print 'cannot process alias %s -> %s' % (als, trg)
|
|
continue
|
|
result[als_seq] = trg_seq
|
|
return result
|
|
|
|
|
|
def add_aliases(filedir, prefix, ext, replace=False, dry_run=False):
|
|
if not path.isdir(filedir):
|
|
print >> sys.stderr, '%s is not a directory' % filedir
|
|
return
|
|
|
|
prefix_len = len(prefix)
|
|
suffix_len = len(ext) + 1
|
|
filenames = [path.basename(f)
|
|
for f in glob.glob(path.join(filedir, '%s*.%s' % (prefix, ext)))]
|
|
seq_to_file = {
|
|
str_to_seq(name[prefix_len:-suffix_len]) : name
|
|
for name in filenames}
|
|
|
|
aliases = read_emoji_aliases()
|
|
aliases_to_create = {}
|
|
aliases_to_replace = []
|
|
for als,trg in sorted(aliases.items()):
|
|
if trg not in seq_to_file:
|
|
print >> sys.stderr, 'target %s for %s does not exist' % (
|
|
seq_to_str(trg), seq_to_str(als))
|
|
continue
|
|
if als in seq_to_file:
|
|
if replace:
|
|
aliases_to_replace.append(seq_to_file[als])
|
|
else:
|
|
print >> sys.stderr, 'alias %s exists' % seq_to_str(als)
|
|
continue
|
|
target_file = seq_to_file[trg]
|
|
alias_name = '%s%s.%s' % (prefix, seq_to_str(als), ext)
|
|
aliases_to_create[alias_name] = target_file
|
|
|
|
if replace:
|
|
if not dry_run:
|
|
for k in sorted(aliases_to_replace):
|
|
os.remove(path.join(filedir, k))
|
|
print 'replacing %d files' % len(aliases_to_replace)
|
|
|
|
for k, v in sorted(aliases_to_create.items()):
|
|
if dry_run:
|
|
msg = 'replace ' if k in aliases_to_replace else ''
|
|
print '%s%s -> %s' % (msg, k, v)
|
|
else:
|
|
try:
|
|
os.symlink(v, path.join(filedir, k))
|
|
except:
|
|
print >> sys.stderr, 'failed to create %s -> %s' % (k, v)
|
|
raise Exception('oops')
|
|
print 'created %d symlinks' % len(aliases_to_create)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'-d', '--filedir', help='directory containing files to alias',
|
|
required=True, metavar='dir')
|
|
parser.add_argument(
|
|
'-p', '--prefix', help='file name prefix (default emoji_u)',
|
|
metavar='pfx', default='emoji_u')
|
|
parser.add_argument(
|
|
'-e', '--ext', help='file name extension (default png)',
|
|
choices=['ai', 'png', 'sgv'], default='png')
|
|
parser.add_argument(
|
|
'-r', '--replace', help='replace existing files/aliases',
|
|
action='store_true')
|
|
parser.add_argument(
|
|
'-n', '--dry_run', help='print out aliases to create only',
|
|
action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
add_aliases(args.filedir, args.prefix, args.ext, args.replace, args.dry_run)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|