|
4 | 4 | Example code to call Rosette API to get the category of a document (at a given URL). |
5 | 5 | """ |
6 | 6 |
|
7 | | -import argparse |
8 | 7 | import json |
9 | 8 |
|
10 | 9 | from rosette.api import API, DocumentParameters |
11 | 10 |
|
12 | | -parser = argparse.ArgumentParser(description="Get the category of a piece of a document at a URL") |
13 | | -parser.add_argument("--key", required=True, help="Rosette API key") |
14 | | -parser.add_argument("--service_url", nargs="?", help="Optional user service URL") |
15 | | -parser.add_argument("--url", nargs="?", default="https://en.wikipedia.org/wiki/Basis_Technology_Corp.", help="Optional URL for data") |
16 | | -args = parser.parse_args() |
17 | 11 |
|
18 | | -# Create an API instance |
19 | | -if args.service_url: |
20 | | - api = API(service_url=args.service_url, user_key=args.key) |
21 | | -else: |
22 | | - api = API(user_key=args.key) |
| 12 | +def run(key): |
| 13 | + url = "https://en.wikipedia.org/wiki/Basis_Technology_Corp." |
| 14 | + # Create an API instance |
| 15 | + api = API(user_key=key) |
| 16 | + params = DocumentParameters() |
23 | 17 |
|
24 | | -params = DocumentParameters() |
| 18 | + # Use a URL to input data instead of a string |
| 19 | + params["contentUri"] = url |
| 20 | + result = api.categories(params) |
25 | 21 |
|
26 | | -# Use a URL to input data instead of a string |
27 | | -params["contentUri"] = args.url |
28 | | - |
29 | | -result = api.categories(params) |
30 | | - |
31 | | -print(json.dumps(result, indent=2, ensure_ascii=False).encode("utf8")) |
| 22 | + print(json.dumps(result, indent=2, ensure_ascii=False).encode("utf8")) |
| 23 | + return json.dumps(result, indent=2, ensure_ascii=False).encode("utf8") |
0 commit comments