# GIS4WRF (https://doi.org/10.5281/zenodo.1288569)

# Copyright (c) 2018 D. Meyer and M. Riechert. Licensed under MIT.

import os

import sys

import platform

import subprocess

import glob

from fnmatch import fnmatch

import zipfile

import shutil

import json

import PyQt5.pyrcc_main as pyrcc

import yaml

PKG_NAME = 'gis4wrf'

THIS_DIR = os.path.abspath(os.path.dirname(__file__))

PKG_DIR = os.path.join(THIS_DIR, PKG_NAME)

ZIP_FILE = PKG_DIR + '.zip'

# All entries will be interpreted with wildcards around them

ZIP_EXCLUDES = [

]

# Create QT resources file

QT_RESOURCES_DIR = os.path.join(PKG_DIR, 'plugin', 'resources')

QT_RESOURCES = glob.glob(os.path.join(QT_RESOURCES_DIR, '*.qrc'))

PYQT_RESOURCE_FILE = os.path.join(QT_RESOURCES_DIR, 'resources.py')

if not pyrcc.processResourceFile(QT_RESOURCES, PYQT_RESOURCE_FILE, listFiles=False):

print('Error creating PyQT resources file', file=sys.stderr)

sys.exit(1)

# Convert YAML to JSON

YAML_DIR = os.path.join(PKG_DIR, 'core', 'readers', 'nml_schemas')

YAML_PATHS = glob.glob(os.path.join(YAML_DIR, '*.yml'))

JSON_PATHS = [os.path.splitext(yaml_path)[0] + '.json' for yaml_path in YAML_PATHS]

for yaml_path, json_path in zip(YAML_PATHS, JSON_PATHS):

with open(yaml_path, 'r') as yaml_file:

with open(json_path, 'w') as json_file:

json.dump(yaml.safe_load(yaml_file), json_file, indent=1)

# Symlink plugin into QGIS plugins folder

if platform.system() == 'Windows' or platform.system() == 'Darwin':

PLATFORM = platform.system()

HOME = os.path.expanduser('~')

if PLATFORM == 'Windows':

QGIS_PLUGINS_DIR = os.path.join(HOME, 'AppData\Roaming\QGIS\QGIS3\profiles\default\python\plugins')

elif PLATFORM == 'Darwin':

QGIS_PLUGINS_DIR = os.path.join(HOME, 'Library/Application Support/QGIS/QGIS3/profiles/default/python/plugins')

QGIS_PLUGIN_DIR = os.path.join(QGIS_PLUGINS_DIR, PKG_NAME)

if not os.path.exists(os.path.dirname(QGIS_PLUGINS_DIR)):

print('Could not find QGIS3 user profile folder, not creating symlink for plugin', file=sys.stderr)

elif os.path.exists(QGIS_PLUGIN_DIR):

print('Folder {} exists, not creating symlink'.format(QGIS_PLUGIN_DIR))

else:

# QGIS only creates the /plugins folder after installing the first plugin

if not os.path.exists(QGIS_PLUGINS_DIR):

os.mkdir(QGIS_PLUGINS_DIR)

try:

if PLATFORM == 'Windows':

subprocess.check_output(['mklink', '/j', QGIS_PLUGIN_DIR, PKG_DIR], shell=True)

elif PLATFORM == 'Darwin':

os.symlink(PKG_DIR, QGIS_PLUGIN_DIR)

except subprocess.CalledProcessError as err:

print('Error creating symlink for plugin:')

print(err.output, file=sys.stderr)

# Copy CHANGELOG.txt, ATTRIBUTION.txt, and LICENSE.txt into gis4wrf plugin directory

print('Copying CHANGELOG.txt to the package directory: '+ PKG_DIR)

shutil.copy(os.path.join(THIS_DIR,'CHANGELOG.txt' ), os.path.join(PKG_DIR,'CHANGELOG.txt'))

print('Copying ATTRIBUTION.txt to the package directory: '+ PKG_DIR)

shutil.copy(os.path.join(THIS_DIR,'ATTRIBUTION.txt' ), os.path.join(PKG_DIR,'ATTRIBUTION.txt'))

print('Copying LICENSE.txt to the package directory: '+ PKG_DIR)

shutil.copy(os.path.join(THIS_DIR,'LICENSE.txt' ), os.path.join(PKG_DIR,'LICENSE.txt'))

# Package into ZIP file

def create_zip(zip_path, folder_path, ignore_patterns):

print('Created ZIP archive ' + zip_path)

create_zip(ZIP_FILE, PKG_DIR, ZIP_EXCLUDES)