marseymoji/add_aliases.py

#!/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_default_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()