small fixes + new api point to get companies of a given APE code

Author: leonard

.gitignore

@@ -116,6 +116,6 @@ notebooks
 venv
 data
 *.tmp
-
+bundles/
 sql_data
 migrations

README.rst

@@ -24,12 +24,16 @@ Create and activate virtual environment python 3.9.4.
 The package uses [pyenv-virtualenv](https://github.com/pyenv/pyenv-virtualenv).
 ```
 make create_environment
+```
+```
 make dev_requirements
 ```
 
 In production run the following instead :
 ```
 make create_environment
+```
+```
 make prod_requirements
 ```
 

enthic/app.py

@@ -14,6 +14,7 @@
     get_financial_data_by_siren,
 )
 from enthic.database.fetch import fetchall
+from enthic.database.mysql import initialize_mysql
 from enthic.decorator.insert_request import insert_request
 from enthic.ontology import (
     APE_CODE,
@@ -50,6 +51,9 @@
 application.config["MYSQL_DB"] = "enthic"
 
 
+initialize_mysql(application)
+
+
 @application.route(
     "/company/siren/<int:siren>/<string:year>", methods=["GET"], strict_slashes=False
 )
@@ -83,6 +87,29 @@ def company_siren(siren):
     return AllSirenCompany(siren)
 
 
+@application.route("/company/ape/<string:ape>", methods=["GET"], strict_slashes=False)
+@insert_request
+def company_ape(ape):
+    sql_query = """SELECT identity.siren, denomination, value FROM identity
+                INNER JOIN annual_statistics ON identity.siren=annual_statistics.siren
+                WHERE stats_type = 1"""
+    enthic_ape_code = get_corresponding_ape_codes(ape)
+
+    if len(enthic_ape_code) > 1:
+        sql_query += f" AND ape IN {tuple(enthic_ape_code)}"
+    elif len(enthic_ape_code) == 1:
+        sql_query += f" AND ape = {enthic_ape_code[0]}"
+    sql_query += """GROUP BY siren
+                    ORDER BY value DESC"""
+    sql_param = {"ape": enthic_ape_code}
+    companies = fetchall(sql_query, sql_param)
+    result = [
+        {"siren": item[0], "denomination": item[1], "score": item[2]}
+        for item in companies
+    ]
+    return OKJSONResponse(result)
+
+
 @application.route("/exist/siren/<sirens>", methods=["GET"], strict_slashes=False)
 def company_denomination_year(sirens):
     """

enthic/database/mysql.py

@@ -1,7 +1,11 @@
-from flask import current_app as application
 from flask_mysqldb import MySQL
 
-try:
-    mysql = MySQL(application)
-except RuntimeError:
-    pass  # RUNTIME ERROR: WORKING OUTSIDE OF APPLICATION CONTEXT. LIKE EXECUTING SPHINX
+mysql = None
+
+
+def initialize_mysql(application):
+    global mysql
+    try:
+        mysql = MySQL(application)
+    except RuntimeError:
+        pass  # RUNTIME ERROR: WORKING OUTSIDE OF APPLICATION CONTEXT. LIKE EXECUTING SPHINX

references/naf_transition.json

@@ -1,6 +1,7 @@
 {
   "65.06": "45.20A",
   "79.06": "68.31Z",
+  "07.11Z": "77.11A",
   "07.42C": "71.12B",
   "07.03D": "68.32A",
   "07.22A": "58.29A",