#!/usr/bin/python
# coding: utf-8

#
# usage: python me.py [-h] -t tag1 -t tag2 file1 file2 ...
#

import sys, os, argparse
from subprocess import Popen, PIPE

applescript = r'''
use AppleScript version "2.7"
use scripting additions
use framework "Foundation"

property OSX : current application
property nil : missing value

on run (argv)
    set n to item 1 of argv

    set argv to items 2 thru end of argv
    set tags to my uniq(items 1 thru n of argv)
    set res to ""

    repeat with f in items (n + 1) thru end of argv
        set fpath to contents of f
        set res to res & my set_tags(tags, fpath) & linefeed
    end repeat
    
    return res
end run

on set_tags(tags, fpath)
    set furl to OSX's NSURL's fileURLWithPath:fpath
    set tkey to OSX's NSURLTagNamesKey
    set tags to OSX's NSArray's arrayWithArray:tags

    set {res, err} to furl's setResourceValue:tags forKey:tkey |error|:(reference)

    if (res) then
        return fpath
    else
        return fpath & ": " & err's localizedDescription as text
    end if
end set_tags

on uniq(tags)
    set new_tags to {}
    repeat with t in tags
        if contents of t is not in new_tags then
            set end of new_tags to contents of t
        end if
    end repeat
    return new_tags
end uniq
'''[1:-1]

def set_tags(tags, args):
    cmd = ['osascript', '-e', applescript, '--', str(len(tags))] + tags + args
    proc = Popen(cmd, stdin = None, stdout = PIPE, stderr = PIPE)

    out, err = proc.communicate()

    if proc.returncode != 0:
        sys.exit(err.rstrip())

    return out.rstrip()

def parse_arguments():
    parser = argparse.ArgumentParser(
        description = 'set tags',
        usage = 'python %(prog)s [-h] -t tag1 -t tag2 -t tag3 file1 file2 ...'
    )

    parser.add_argument(
        '-t', '--tag', required = True,
        action  = 'append',
        dest    = 'tags',
        metavar = 'tags',
        help    = 'specify tag name'
    )

    parser.add_argument(
        'files',
        nargs = '+',
        help  = 'file1 file2 ...'
    )

    args = parser.parse_args()

    return (args.tags, args.files)

def main():
    if len(sys.argv) < 2:
        sys.exit('usage: python {0} file1 file2 ...'.format(__file__))

    tags, args = parse_arguments()

    tags = [x for x in tags if x]
    args = [x for x in args if os.path.exists(x)]

    res = set_tags(tags, args)
    print(res)

main()