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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
#!/usr/bin/env python3
"""
Note: this is *not* the tool used to generate "official" metadata dumps; that
tool is written in rust and runs on the production infrastructure for speed.
These scripts are just a demonstration of how the API *could* be scraped
without permission by an third party.
"""
import sys
import json
import argparse
import fatcat_client
from fatcat_client.rest import ApiException
from fatcat_client import ReleaseEntity, ContainerEntity, ChangelogEntry
from fatcat_tools import uuid2fcid, entity_from_json, entity_to_dict, \
release_to_elasticsearch, container_to_elasticsearch, \
changelog_to_elasticsearch, public_api
def run_export_releases(args):
api = args.api
for line in args.ident_file:
ident = uuid2fcid(line.split()[0])
release = api.get_release(ident=ident, expand="all")
args.json_output.write(
json.dumps(entity_to_dict(release)) + "\n")
def run_transform_releases(args):
for line in args.json_input:
line = line.strip()
if not line:
continue
entity = entity_from_json(line, ReleaseEntity)
args.json_output.write(
json.dumps(release_to_elasticsearch(entity)) + '\n')
def run_transform_containers(args):
for line in args.json_input:
line = line.strip()
if not line:
continue
entity = entity_from_json(line, ContainerEntity)
args.json_output.write(
json.dumps(container_to_elasticsearch(entity)) + '\n')
def run_transform_changelogs(args):
for line in args.json_input:
line = line.strip()
if not line:
continue
entity = entity_from_json(line, ChangelogEntry)
args.json_output.write(
json.dumps(changelog_to_elasticsearch(entity)) + '\n')
def run_export_changelog(args):
api = args.api
end = args.end
if end is None:
latest = api.get_changelog(limit=1)[0]
end = latest.index
for i in range(args.start, end):
entry = api.get_changelog_entry(index=i)
args.json_output.write(
json.dumps(entity_to_dict(entry)) + "\n")
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--debug',
action='store_true',
help="enable debugging interface")
parser.add_argument('--host-url',
default="http://localhost:9411/v0",
help="connect to this host/port")
subparsers = parser.add_subparsers()
sub_releases = subparsers.add_parser('releases')
sub_releases.set_defaults(func=run_export_releases)
sub_releases.add_argument('ident_file',
help="TSV list of fatcat release idents to dump",
default=sys.stdin, type=argparse.FileType('r'))
sub_releases.add_argument('json_output',
help="where to send output",
default=sys.stdout, type=argparse.FileType('w'))
sub_transform_releases = subparsers.add_parser('transform-releases')
sub_transform_releases.set_defaults(func=run_transform_releases)
sub_transform_releases.add_argument('json_input',
help="JSON-per-line of release entities",
default=sys.stdin, type=argparse.FileType('r'))
sub_transform_releases.add_argument('json_output',
help="where to send output",
default=sys.stdout, type=argparse.FileType('w'))
sub_transform_containers = subparsers.add_parser('transform-containers')
sub_transform_containers.set_defaults(func=run_transform_containers)
sub_transform_containers.add_argument('json_input',
help="JSON-per-line of container entities",
default=sys.stdin, type=argparse.FileType('r'))
sub_transform_containers.add_argument('json_output',
help="where to send output",
default=sys.stdout, type=argparse.FileType('w'))
sub_transform_changelogs = subparsers.add_parser('transform-changelogs')
sub_transform_changelogs.set_defaults(func=run_transform_changelogs)
sub_transform_changelogs.add_argument('json_input',
help="JSON-per-line of changelog entries",
default=sys.stdin, type=argparse.FileType('r'))
sub_transform_changelogs.add_argument('json_output',
help="where to send output",
default=sys.stdout, type=argparse.FileType('w'))
sub_changelog = subparsers.add_parser('changelog')
sub_changelog.set_defaults(func=run_export_changelog)
sub_changelog.add_argument('--start',
help="index to start dumping at",
default=1, type=int)
sub_changelog.add_argument('--end',
help="index to stop dumping at (else detect most recent)",
default=None, type=int)
sub_changelog.add_argument('json_output',
help="where to send output",
default=sys.stdout, type=argparse.FileType('w'))
args = parser.parse_args()
if not args.__dict__.get("func"):
print("tell me what to do!")
sys.exit(-1)
args.api = public_api(args.host_url)
args.func(args)
if __name__ == '__main__':
main()
|
