python-tika/setup.py at master · sudharsh/python-tika · GitHub

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
import os
import sys
import os.path

from distutils import errors
from jcc import cpp

TIKA_VERSION = 1.1
TIKA_APP = "tika-app-%s.jar" %(TIKA_VERSION)
TIKA_CORE = "tika-core-%s.jar" %(TIKA_VERSION)
TIKA_PARSERS = "tika-parsers-%s.jar" %(TIKA_VERSION)

deps =  {
    "osgi": "org.eclipse.osgi.jar",
    "tika_app": TIKA_APP,
    "tika_core": TIKA_CORE,
    "tika_parsers": TIKA_PARSERS,
    "log4j": "log4j.properties.jar",
    "jcc": "jcc"
    }


def check_deps(required, found):
    satisfied = True
    for r in required:
        if r not in found:
            satisfied = False
            print "%s not found!" % (required[r])
    return satisfied


def find_deps(required):
    classpath = [x for x in os.environ.get("CLASSPATH", "").split(":") if x] + ["lib", "win", os.getcwdu()]
    path = sys.path + ["lib", "win"]
    found = {}
    def __probe(paths):
        for p in paths:
            for j in required:
                joined = os.path.join(p, deps[j])
                if os.path.exists(joined) and not j in found:
                    found[j] = joined

    __probe(classpath)
    __probe(path)
    return found


def get_jcc_args(jcc_path):
    jcc_options = {
        'include': (found['osgi'],  found['log4j'], found['tika_app']),
        'package': ('org.xml.sax', 'org.apache.tika'),
        'jar': (found['tika_parsers'], found['tika_core']),
        'python': 'tika',
        'version': TIKA_VERSION,
        'module': 'parser',
        'reserved': ('asm',),
        'classes': ('java.io.File', 'java.io.FileInputStream', 'java.io.ByteArrayInputStream'),
        }
    jcc_args = [os.path.join(jcc_path, 'nonexistent-argv-0')]
    setup_args = []
    egg_info_mode = False
    maxheap = '64m'
    for k, v in jcc_options.iteritems():
        if k in ['classes']:
            jcc_args.extend(v)
        elif hasattr(v, '__iter__'):
            for value in v:
                jcc_args.append('--%s' % k)
                jcc_args.append(value)
        else:
            jcc_args.append('--%s' % k)
            jcc_args.append(v)

    for arg in sys.argv[1:]:
        if arg == 'install':
            jcc_args.append('--install')
        elif arg == 'build':
            jcc_args.append('--build')
        elif arg == '-c':
            pass
        elif arg == 'egg_info':
            jcc_args.append('--egg-info')
        elif arg == '--vmarg':
            jcc_args.append(arg)
            i += 1
            jcc_args.append(sys.argv[i])
        elif arg == '--maxheap':
            i += 1
            maxheap = sys.argv[i]
        else:
            setup_args.append(arg)

    for extra_arg in setup_args:
        jcc_args.append('--extra-setup-arg')
        jcc_args.append(extra_arg)

    jcc_args.extend(['--maxheap', maxheap])
    return jcc_args


found = find_deps(deps)
if not check_deps(deps, found):
    raise errors.DistutilsFileError("Dependencies not satisfied")
jcc_args = get_jcc_args(found["jcc"])
cpp.jcc(jcc_args)