Move mongodb migrations inside the server's folder

This commit is contained in:
allegroai 2019-12-29 08:58:54 +02:00
parent b536020058
commit 551be12f01
4 changed files with 12 additions and 6 deletions

View File

@ -10,7 +10,11 @@ from pathlib import Path
from requests.adapters import HTTPAdapter
from requests.packages.urllib3.util.retry import Retry
HERE = Path(__file__).parent
HERE = Path(__file__).resolve().parent
session = requests.Session()
adapter = HTTPAdapter(max_retries=Retry(5, backoff_factor=0.5))
session.mount('http://', adapter)
def apply_mappings_to_host(host: str):
@ -20,10 +24,6 @@ def apply_mappings_to_host(host: str):
es_server = host
url = f"{es_server}/_template/{f.stem}"
session = requests.Session()
adapter = HTTPAdapter(max_retries=Retry(5, backoff_factor=0.5))
session.mount('http://', adapter)
session.delete(url)
r = session.post(
url,

View File

@ -25,7 +25,7 @@ from service_repo.auth.fixed_user import FixedUser
log = config.logger(__file__)
migration_dir = (Path(__file__) / "../../migration/mongodb").resolve()
migration_dir = Path(__file__).resolve().parent / "mongo" / "migrations"
class MissingElasticConfiguration(Exception):

View File

@ -0,0 +1,6 @@
from pymongo.database import Database, Collection
def migrate_auth(db: Database):
collection: Collection = db["user"]
collection.drop_index("name_1_company_1")