mirror of
https://gitlab.com/wgp/dougal/software.git
synced 2025-12-06 11:07:08 +00:00
58 lines
1.4 KiB
Python
Executable File
58 lines
1.4 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
"""
|
|
Import preplot data.
|
|
|
|
For each preplot file defined in each survey, check for new
|
|
or modified preplots and (re-)import them into the database.
|
|
"""
|
|
|
|
from glob import glob
|
|
import os
|
|
import sys
|
|
import time
|
|
import configuration
|
|
import preplots
|
|
from datastore import Datastore
|
|
|
|
if __name__ == '__main__':
|
|
|
|
print("Connecting to database")
|
|
db = Datastore()
|
|
surveys = db.surveys()
|
|
|
|
print("Reading configuration")
|
|
file_min_age = configuration.read().get('imports', {}).get('file_min_age', 10)
|
|
|
|
print("Reading surveys")
|
|
for survey in surveys:
|
|
print(f'Survey: {survey["id"]} ({survey["schema"]})')
|
|
db.set_survey(survey["schema"])
|
|
for file in survey["preplots"]:
|
|
realpath = configuration.translate_path(file["path"])
|
|
|
|
print(f"Preplot: {file['path']}")
|
|
if not db.file_in_db(file["path"]):
|
|
|
|
age = time.time() - os.path.getmtime(realpath)
|
|
if age < file_min_age:
|
|
print("Skipping file because too new", file["path"])
|
|
continue
|
|
|
|
print("Importing")
|
|
try:
|
|
preplot = preplots.from_file(file, realpath)
|
|
except FileNotFoundError:
|
|
print(f"File does not exist: {file['path']}", file=sys.stderr)
|
|
continue
|
|
|
|
if type(preplot) is list:
|
|
print("Saving to DB")
|
|
db.save_preplots(preplot, file["path"], file["class"], survey["epsg"], file)
|
|
elif type(preplot) is str:
|
|
print(preplot)
|
|
else:
|
|
print("Already in DB")
|
|
|
|
print("Done")
|