# -*- coding: utf-8 -*-
from setuptools import setup

packages = \
['udata_hydra',
 'udata_hydra.analysis',
 'udata_hydra.crawl',
 'udata_hydra.db',
 'udata_hydra.migrations',
 'udata_hydra.routes',
 'udata_hydra.schemas',
 'udata_hydra.utils']

package_data = \
{'': ['*'], 'udata_hydra.migrations': ['csv/*', 'main/*']}

install_requires = \
['aiocontextvars>=0.2.2,<0.3.0',
 'aiohttp>=3.10.3,<4.0.0',
 'asyncpg>=0.29.0,<0.30.0',
 'coloredlogs>=15.0.1,<16.0.0',
 'csv-detective==0.7.3',
 'dateparser>=1.1.7,<2.0.0',
 'humanfriendly>=10.0,<11.0',
 'marshmallow>=3.14.1,<4.0.0',
 'minicli>=0.5.3,<0.6.0',
 'minio==7.2.7',
 'progressist>=0.1.0,<0.2.0',
 'pyarrow==16.1.0',
 'python-dateutil>=2.8.2,<3.0.0',
 'python-magic>=0.4.25,<0.5.0',
 'python-slugify>=8.0.4,<9.0.0',
 'redis>=4.1.4,<5.0.0',
 'rq>=1.11.1,<2.0.0',
 'sentry-sdk>=2.10.0,<3.0.0',
 'sqlalchemy>=1.4.46,<2.0.0',
 'str2bool>=1.1,<2.0',
 'str2float>=0.0.9,<0.0.10']

entry_points = \
{'console_scripts': ['udata-hydra = udata_hydra.cli:run',
                     'udata-hydra-app = udata_hydra.app:run',
                     'udata-hydra-crawl = udata_hydra.crawl:run']}

setup_kwargs = {
    'name': 'udata-hydra',
    'version': '2.1.1.dev6764',
    'description': 'Async crawler and parsing service for data.gouv.fr',
    'long_description': '# udata-hydra 🦀\n\n`udata-hydra` is an async metadata crawler for [data.gouv.fr](https://www.data.gouv.fr).\n\nURLs are crawled via _aiohttp_, catalog and crawled metadata are stored in a _PostgreSQL_ database.\n\nSince it\'s called _hydra_, it also has mythical powers embedded:\n- analyse remote resource metadata over time to detect changes in the smartest way possible\n- if the remote resource is a CSV, convert it to a PostgreSQL table, ready for APIfication\n- send crawl and analysis info to a udata instance\n\n## Architecture schema\n\nThe architecture for the full workflow is the following:\n\n![Full workflow architecture](docs/archi-idd-IDD.drawio.png)\n\n\nThe hydra crawler is one of the components of the architecture. It will check if resource is available, analyse the type of file if the resource has been modified, and analyse the CSV content. It will also convert CSV resources to database tables and send the data to a udata instance.\n\n![Crawler architecture](docs/hydra.drawio.png)\n\n## Dependencies\n\nThis project uses `libmagic`, which needs to be installed on your system, eg:\n\n`brew install libmagic` on MacOS, or `sudo apt-get install libmagic-dev` on linux.\n\n## CLI\n\n### Create database structure\n\nInstall udata-hydra dependencies and cli.\n`poetry install`\n\n`poetry run udata-hydra migrate`\n\n### Load (UPSERT) latest catalog version from data.gouv.fr\n\n`poetry run udata-hydra load-catalog`\n\n## Crawler\n\n`poetry run udata-hydra-crawl`\n\nIt will crawl (forever) the catalog according to the config set in `config.toml`, with a default config in `udata_hydra/config_default.toml`.\n\n`BATCH_SIZE` URLs are queued at each loop run.\n\nThe crawler will start with URLs never checked and then proceed with URLs crawled before `CHECK_DELAYS` interval. It will then wait until something changes (catalog or time).\n\nThere\'s a by-domain backoff mecanism. The crawler will wait when, for a given domain in a given batch, `BACKOFF_NB_REQ` is exceeded in a period of `BACKOFF_PERIOD` seconds. It will retry until the backoff is lifted.\n\nIf an URL matches one of the `EXCLUDED_PATTERNS`, it will never be checked.\n\n## Worker\n\nA job queuing system is used to process long-running tasks. Launch the worker with the following command:\n\n`poetry run rq worker -c udata_hydra.worker`\n\nTo monitor worker status:\n\n`poetry run rq info -c udata_hydra.worker --interval 1`\n\nTo empty all the queues:\n\n`poetry run rq empty -c udata_hydra.worker low default high`\n\n## CSV conversion to database\n\nConverted CSV tables will be stored in the database specified via `config.DATABASE_URL_CSV`. For tests it\'s same database as for the catalog. Locally, `docker compose` will launch two distinct database containers.\n\n## Tests\n\nTo run the tests, you need to launch the database, the test database, and the Redis broker with `docker compose -f docker-compose.yml -f docker-compose.test.yml -f docker-compose.broker.yml up -d`.\n\nThen you can run the tests with `poetry run pytest`.\n\nTo run a specific test file, you can pass the path to the file to pytest, like this: `poetry run pytest tests/test_app.py`.\n\nTo run a specific test function, you can pass the path to the file and the name of the function to pytest, like this: `poetry run pytest tests/test_app.py::test_get_latest_check`.\n\nIf you would like to see print statements as they are executed, you can pass the -s flag to pytest (`poetry run pytest -s`). However, note that this can sometimes be difficult to parse.\n\n### Tests coverage\n\nPytest automatically uses the `coverage` package to generate a coverage report, which is displayed at the end of the test run in the terminal.\nThe coverage is configured in the `pypoject.toml` file, in the `[tool.pytest.ini_options]` section.\nYou can also override the coverage report configuration when running the tests by passing some flags like `--cov-report` to pytest. See [the pytest-cov documentation](https://pytest-cov.readthedocs.io/en/latest/config.html) for more information.\n\n## API\n\nThe API will need a Bearer token for each request on protected endpoints (any endpoint that isn\'t a `GET`).\nThe token is configured in the `config.toml` file as `API_KEY`, and has a default value set in the `udata_hydra/config_default.toml` file.\n\nIf you\'re using hydra as an external service to receive resource events from [udata](https://github.com/opendatateam/udata), then udata needs to also configure this\nAPI key in its `udata.cfg` file:\n\n```python\n# Wether udata should publish the resource events\nPUBLISH_ON_RESOURCE_EVENTS = True\n# Where to publish the events\nRESOURCES_ANALYSER_URI = "http://localhost:8000"\n# The API key that hydra needs\nRESOURCES_ANALYSER_API_KEY = "api_key_to_change"\n```\n\n### Run\n\n```bash\npoetry install\npoetry run adev runserver udata_hydra/app.py\n```\nBy default, the app will listen on `localhost:8000`.\nYou can check the status of the app with `curl http://localhost:8000/api/health`.\n\n### Routes/endpoints\n\nThe API serves the following endpoints:\n\n*Related to checks:*\n- `GET` on `/api/checks/latest?url={url}&resource_id={resource_id}` to get the latest check for a given URL and/or `resource_id`\n- `GET` on `/api/checks/all?url={url}&resource_id={resource_id}` to get all checks for a given URL and/or `resource_id`\n- `GET` on `/api/checks/aggregate?group_by={column}&created_at={date}` to get checks occurences grouped by a `column` for a specific `date`\n\n*Related to resources:*\n- `GET` on `/api/resources/{resource_id}` to get a resource in the DB "catalog" table from its `resource_id`\n- `POST` on `/api/resources` to receive a resource creation event from a source. It will create a new resource in the DB "catalog" table and mark it as priority for next crawling\n- `PUT` on `/api/resources/{resource_id}` to update a resource in the DB "catalog" table\n- `DELETE` on `/api/resources/{resource_id}` to delete a resource in the DB "catalog" table\n\n> :warning: **Warning: the following routes are deprecated and need be removed in the future:**\n> - `POST` on `/api/resource/created` -> use `POST` on `/api/resources/` instead\n> - `POST` on `/api/resource/updated` -> use `PUT` on `/api/resources/` instead\n> - `POST` on `/api/resource/deleted` -> use `DELETE` on `/api/resources/` instead\n\n*Related to resources exceptions:*\n- `GET` on `/api/resources-exceptions` to get the list all resources exceptions\n- `POST` on `/api/resources-exceptions` to create a new resource exception in the DB\n- `PUT` on `/api/resources-exceptions/{resource_id}` to update a resource exception in the DB\n- `DELETE` on `/api/resources-exceptions/{resource_id}` to delete a resource exception from the DB\n\n*Related to some status and health check:*\n- `GET` on `/api/status/crawler` to get the crawling status\n- `GET` on `/api/status/worker` to get the worker status\n- `GET` on `/api/stats` to get the crawling stats\n- `GET` on `/api/health` to get the API version number and environment\n\nYou may want to you a helper such as [Bruno](https://www.usebruno.com/) to handle API calls, in which case all the endpoints are ready to use [here](https://github.com/datagouv/api-calls).\nMore details about some enpoints are provided below with examples, but not for all of them:\n\n#### Get latest check\n\nWorks with `?url={url}` and `?resource_id={resource_id}`.\n\n```bash\n$ curl -s "http://localhost:8000/api/checks/latest?url=http://opendata-sig.saintdenis.re/datasets/661e19974bcc48849bbff7c9637c5c28_1.csv" | json_pp\n{\n   "status" : 200,\n   "catalog_id" : 64148,\n   "deleted" : false,\n   "error" : null,\n   "created_at" : "2021-02-06T12:19:08.203055",\n   "response_time" : 0.830198049545288,\n   "url" : "http://opendata-sig.saintdenis.re/datasets/661e19974bcc48849bbff7c9637c5c28_1.csv",\n   "domain" : "opendata-sig.saintdenis.re",\n   "timeout" : false,\n   "id" : 114750,\n   "dataset_id" : "5c34944606e3e73d4a551889",\n   "resource_id" : "b3678c59-5b35-43ad-9379-fce29e5b56fe",\n   "headers" : {\n      "content-disposition" : "attachment; filename=\\"xn--Dlimitation_des_cantons-bcc.csv\\"",\n      "server" : "openresty",\n      "x-amz-meta-cachetime" : "191",\n      "last-modified" : "Wed, 29 Apr 2020 02:19:04 GMT",\n      "content-encoding" : "gzip",\n      "content-type" : "text/csv",\n      "cache-control" : "must-revalidate",\n      "etag" : "\\"20415964703d9ccc4815d7126aa3a6d8\\"",\n      "content-length" : "207",\n      "date" : "Sat, 06 Feb 2021 12:19:08 GMT",\n      "x-amz-meta-contentlastmodified" : "2018-11-19T09:38:28.490Z",\n      "connection" : "keep-alive",\n      "vary" : "Accept-Encoding"\n   }\n}\n```\n\n#### Get all checks for an URL or resource\n\nWorks with `?url={url}` and `?resource_id={resource_id}`.\n\n```bash\n$ curl -s "http://localhost:8000/api/checks/all?url=http://www.drees.sante.gouv.fr/IMG/xls/er864.xls" | json_pp\n[\n   {\n      "domain" : "www.drees.sante.gouv.fr",\n      "dataset_id" : "53d6eadba3a72954d9dd62f5",\n      "timeout" : false,\n      "deleted" : false,\n      "response_time" : null,\n      "error" : "Cannot connect to host www.drees.sante.gouv.fr:443 ssl:True [SSLCertVerificationError: (1, \\"[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: Hostname mismatch, certificate is not valid for \'www.drees.sante.gouv.fr\'. (_ssl.c:1122)\\")]",\n      "catalog_id" : 232112,\n      "url" : "http://www.drees.sante.gouv.fr/IMG/xls/er864.xls",\n      "headers" : {},\n      "id" : 165107,\n      "created_at" : "2021-02-06T14:32:47.675854",\n      "resource_id" : "93dfd449-9d26-4bb0-a6a9-ee49b1b8a4d7",\n      "status" : null\n   },\n   {\n      "timeout" : false,\n      "deleted" : false,\n      "response_time" : null,\n      "error" : "Cannot connect to host www.drees.sante.gouv.fr:443 ssl:True [SSLCertVerificationError: (1, \\"[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: Hostname mismatch, certificate is not valid for \'www.drees.sante.gouv.fr\'. (_ssl.c:1122)\\")]",\n      "domain" : "www.drees.sante.gouv.fr",\n      "dataset_id" : "53d6eadba3a72954d9dd62f5",\n      "created_at" : "2020-12-24T17:06:58.158125",\n      "resource_id" : "93dfd449-9d26-4bb0-a6a9-ee49b1b8a4d7",\n      "status" : null,\n      "catalog_id" : 232112,\n      "url" : "http://www.drees.sante.gouv.fr/IMG/xls/er864.xls",\n      "headers" : {},\n      "id" : 65092\n   }\n]\n```\n\n#### Get checks occurences grouped by a column for a specific date\n\nWorks with `?group_by={column}` and `?created_at={date}`.\n`date` should be a date in format `YYYY-MM-DD` or the default keyword `today`.\n\n```bash\n$ curl -s "http://localhost:8000/api/checks/aggregate?group_by=domain&created_at=today" | json_pp\n[\n  {\n    "value": "www.geo2france.fr",\n    "count": 4\n  },\n  {\n    "value": "static.data.gouv.fr",\n    "count": 4\n  },\n  {\n    "value": "grandestprod.data4citizen.com",\n    "count": 3\n  },\n  {\n    "value": "www.datasud.fr",\n    "count": 2\n  },\n  {\n    "value": "koumoul.com",\n    "count": 2\n  },\n  {\n    "value": "opendata.aude.fr",\n    "count": 2\n  },\n  {\n    "value": "departement-ain.opendata.arcgis.com",\n    "count": 2\n  },\n  {\n    "value": "opendata.agglo-larochelle.fr",\n    "count": 1\n  }\n]\n```\n\n#### Adding a resource exception\n\n```bash\n$ curl   -X POST http://localhost:8000/api/resources-exceptions \\\n         -H \'Authorization: Bearer <myAPIkey>\' \\\n         -d \'{\n            "resource_id": "123e4567-e89b-12d3-a456-426614174000",\n            "table_indexes": {\n                  "siren": "index"\n            },\n            "comment": "This is a comment for the resource exception."\n         }\'\n```\n\n...or, if you don\'t want to add table indexes and a comment:\n```bash\n$ curl  -X POST localhost:8000/api/resources-exceptions \\\n        -H \'Authorization: Bearer <myAPIkey>" \\\n        -d \'{"resource_id": "f868cca6-8da1-4369-a78d-47463f19a9a3"}\'\n```\n\n#### Updating a resource exception\n\n```bash\n$ curl   -X PUT http://localhost:8000/api/resources-exceptions/f868cca6-8da1-4369-a78d-47463f19a9a3 \\\n         -H "Authorization: Bearer <myAPIkey>" \\\n         -d \'{\n            "table_indexes": {\n                  "siren": "index",\n                  "code_postal": "index"\n            },\n            "comment": "Updated comment for the resource exception."\n         }\'\n```\n\n#### Deleting a resource exception\n\n```bash\n$ curl  -X DELETE http://localhost:8000/api/resources-exceptions/f868cca6-8da1-4369-a78d-47463f19a9a3 \\\n        -H "Authorization: Bearer <myAPIkey>"\n```\n\n#### Get crawling status\n\n```bash\n$ curl -s "http://localhost:8000/api/status/crawler" | json_pp\n{\n   "fresh_checks_percentage" : 0.4,\n   "pending_checks" : 142153,\n   "total" : 142687,\n   "fresh_checks" : 534,\n   "checks_percentage" : 0.4,\n   "resources_statuses_count": {\n      "null": 195339,\n      "BACKOFF": 0,\n      "CRAWLING_URL": 0,\n      "TO_ANALYSE_RESOURCE": 1,\n      "ANALYSING_RESOURCE": 0,\n      "TO_ANALYSE_CSV": 0,\n      "ANALYSING_CSV": 0,\n      "INSERTING_IN_DB": 0,\n      "CONVERTING_TO_PARQUET": 0\n  }\n}\n```\n\n#### Get worker status\n\n```bash\n$ curl -s "http://localhost:8000/api/status/worker" | json_pp\n{\n   "queued" : {\n      "default" : 0,\n      "high" : 825,\n      "low" : 655\n   }\n}\n```\n\n#### Get crawling stats\n\n```bash\n$ curl -s "http://localhost:8000/api/stats" | json_pp\n{\n   "status" : [\n      {\n         "count" : 525,\n         "percentage" : 98.3,\n         "label" : "ok"\n      },\n      {\n         "label" : "error",\n         "percentage" : 1.3,\n         "count" : 7\n      },\n      {\n         "label" : "timeout",\n         "percentage" : 0.4,\n         "count" : 2\n      }\n   ],\n   "status_codes" : [\n      {\n         "code" : 200,\n         "count" : 413,\n         "percentage" : 78.7\n      },\n      {\n         "code" : 501,\n         "percentage" : 12.4,\n         "count" : 65\n      },\n      {\n         "percentage" : 6.1,\n         "count" : 32,\n         "code" : 404\n      },\n      {\n         "code" : 500,\n         "percentage" : 2.7,\n         "count" : 14\n      },\n      {\n         "code" : 502,\n         "count" : 1,\n         "percentage" : 0.2\n      }\n   ]\n}\n```\n\n## Using Webhook integration\n\n** Set the config values**\n\nCreate a `config.toml` where your service and commands are launched, or specify a path to a TOML file via the `HYDRA_SETTINGS` environment variable. `config.toml` or equivalent will override values from `udata_hydra/config_default.toml`, lookup there for values that can/need to be defined.\n\n```toml\nUDATA_URI = "https://dev.local:7000/api/2"\nUDATA_URI_API_KEY = "example.api.key"\nSENTRY_DSN = "https://{my-sentry-dsn}"\n```\n\nThe webhook integration sends HTTP messages to `udata` when resources are analysed or checked to fill resources extras.\n\nRegarding analysis, there is a phase called "change detection". It will try to guess if a resource has been modified based on different criterions:\n- harvest modified date in catalog\n- content-length and last-modified headers\n- checksum comparison over time\n\nThe payload should look something like:\n\n```json\n{\n   "analysis:content-length": 91661,\n   "analysis:mime-type": "application/zip",\n   "analysis:checksum": "bef1de04601dedaf2d127418759b16915ba083be",\n   "analysis:last-modified-at": "2022-11-27T23:00:54.762000",\n   "analysis:last-modified-detection": "harvest-resource-metadata",\n}\n```\n\n## Development\n\n### docker compose\n\nMultiple docker-compose files are provided:\n- a minimal `docker-compose.yml` with two PostgreSQL containers (one for catalog and metadata, the other for converted CSV to database)\n- `docker-compose.broker.yml` adds a Redis broker\n- `docker-compose.test.yml` launches a test DB, needed to run tests\n\nNB: you can launch compose from multiple files like this: `docker compose -f docker-compose.yml -f docker-compose.test.yml up`\n\n### Logging & Debugging\n\nThe log level can be adjusted using the environment variable LOG_LEVEL.\nFor example, to set the log level to `DEBUG` when initializing the database, use `LOG_LEVEL="DEBUG" udata-hydra init_db `.\n\n### Writing a migration\n\n1. Add a file named `migrations/{YYYYMMDD}_{description}.sql` and write the SQL you need to perform migration.\n2. `udata-hydra migrate` will migrate the database as needeed.\n\n## Deployment\n\n3 services need to be deployed for the full stack to run:\n- worker\n- api / app\n- crawler\n\nRefer to each section to learn how to launch them. The only differences from dev to prod are:\n- use `HYDRA_SETTINGS` env var to point to your custom `config.toml`\n- use `HYDRA_APP_SOCKET_PATH` to configure where aiohttp should listen to a [reverse proxy connection (eg nginx)](https://docs.aiohttp.org/en/stable/deployment.html#nginx-configuration) and use `udata-hydra-app` to launch the app server\n\n## Contributing\n\nBefore contributing to the repository and making any PR, it is necessary to initialize the pre-commit hooks:\n```bash\npre-commit install\n```\nOnce this is done, code formatting and linting, as well as import sorting, will be automatically checked before each commit.\n\nIf you cannot use pre-commit, it is necessary to format, lint, and sort imports with [Ruff](https://docs.astral.sh/ruff/) before committing:\n```bash\nruff check --fix .\nruff format .\n```\n\n### Releases\n\nThe release process uses [bump\'X](https://github.com/datagouv/bumpx).\n',
    'author': 'Opendata Team',
    'author_email': 'opendatateam@data.gouv.fr',
    'maintainer': 'None',
    'maintainer_email': 'None',
    'url': 'None',
    'packages': packages,
    'package_data': package_data,
    'install_requires': install_requires,
    'entry_points': entry_points,
    'python_requires': '==3.11.2',
}


setup(**setup_kwargs)
