Blame view

web/run.py 30 KB
bde97e4d   Goutte   Add more changes ...
1
2
# coding=utf-8

9390ec89   Goutte   Initial experimen...
3
import StringIO
bc18b96c   Goutte   Implement first (...
4
import datetime
8644387c   Goutte   Use real data.
5
import gzip
bc18b96c   Goutte   Implement first (...
6
7
8
import json
import logging
import random
2fedd73b   Goutte   Initial implement...
9
import tarfile
bde97e4d   Goutte   Add more changes ...
10
import time
8644387c   Goutte   Use real data.
11
import urllib
9390ec89   Goutte   Initial experimen...
12
from csv import writer as csv_writer
bc18b96c   Goutte   Implement first (...
13
14
15
16
from math import sqrt
from os import environ, remove as removefile
from os.path import isfile, join, abspath, dirname

9390ec89   Goutte   Initial experimen...
17
from flask import Flask
9390ec89   Goutte   Initial experimen...
18
from flask import request
bc18b96c   Goutte   Implement first (...
19
from flask import url_for, send_from_directory, abort as abort_flask
bde97e4d   Goutte   Add more changes ...
20
from jinja2 import Environment, FileSystemLoader, Markup
57493104   Goutte   Add the time to t...
21
from netCDF4 import Dataset, date2num
bc18b96c   Goutte   Implement first (...
22
from yaml import load as yaml_load
9390ec89   Goutte   Initial experimen...
23
24
25
26
27
28
29

# PATH RELATIVITY #############################################################

THIS_DIRECTORY = dirname(abspath(__file__))


def get_path(relative_path):
a4a9ef03   Goutte   Cache generated C...
30
    """Get an absolute path from the relative path to this script directory."""
9390ec89   Goutte   Initial experimen...
31
32
33
34
35
36
37
38
39
40
41
42
43
    return abspath(join(THIS_DIRECTORY, relative_path))


# COLLECT GLOBAL INFORMATION FROM SOURCES #####################################

# VERSION
with open(get_path('../VERSION'), 'r') as version_file:
    version = version_file.read().strip()

# CONFIG
with open(get_path('../config.yml'), 'r') as config_file:
    config = yaml_load(config_file.read())

c0df94bc   Goutte   Adding more logs.
44
45
FILE_DATE_FMT = "%Y-%m-%dT%H:%M:%S"

9390ec89   Goutte   Initial experimen...
46

f75faf5f   Goutte   WIP
47
48
49
# LOGGING #####################################################################

log = logging.getLogger("HelioPropa")
9bfa6c42   Goutte   More bug hunting.
50
log.setLevel(logging.DEBUG)
077980eb   Goutte   Improve availabil...
51
# log.setLevel(logging.ERROR)                        # <-- set log level here !
b2837a08   Goutte   Add three retries...
52
53
54
55
56
logHandler = logging.FileHandler(get_path('run.log'))
logHandler.setFormatter(logging.Formatter(
    "%(asctime)s - %(levelname)s - %(message)s"
))
log.addHandler(logHandler)
f75faf5f   Goutte   WIP
57
58


9390ec89   Goutte   Initial experimen...
59
60
61
62
# SETUP FLASK ENGINE ##########################################################

app = Flask(__name__, root_path=THIS_DIRECTORY)
app.debug = environ.get('DEBUG') == 'true'
b2837a08   Goutte   Add three retries...
63
if app.debug:
2fedd73b   Goutte   Initial implement...
64
    log.info("Starting Flask app IN DEBUG MODE...")
b2837a08   Goutte   Add three retries...
65
66
else:
    log.info("Starting Flask app...")
9390ec89   Goutte   Initial experimen...
67
68
69
70
71
72
73
74
75
76
77


# SETUP JINJA2 TEMPLATE ENGINE ################################################

def static_global(filename):
    return url_for('static', filename=filename)


def shuffle_filter(seq):
    """
    This shuffles the sequence it is applied to.
2fedd73b   Goutte   Initial implement...
78
    Jinja2 _should_ provide this.
9390ec89   Goutte   Initial experimen...
79
80
81
82
83
84
85
86
87
88
89
    """
    try:
        result = list(seq)
        random.shuffle(result)
        return result
    except:
        return seq


def markdown_filter(value, nl2br=False, p=True):
    """
2fedd73b   Goutte   Initial implement...
90
    Converts markdown into html.
9390ec89   Goutte   Initial experimen...
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
    nl2br: set to True to replace line breaks with <br> tags
    p: set to False to remove the enclosing <p></p> tags
    """
    from markdown import markdown
    from markdown.extensions.nl2br import Nl2BrExtension
    from markdown.extensions.abbr import AbbrExtension
    extensions = [AbbrExtension()]
    if nl2br is True:
        extensions.append(Nl2BrExtension())
    markdowned = markdown(value, output_format='html5', extensions=extensions)
    if p is False:
        markdowned = markdowned.replace(r"<p>", "").replace(r"</p>", "")
    return markdowned


bde97e4d   Goutte   Add more changes ...
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
_js_escapes = {
        '\\': '\\u005C',
        '\'': '\\u0027',
        '"': '\\u0022',
        '>': '\\u003E',
        '<': '\\u003C',
        '&': '\\u0026',
        '=': '\\u003D',
        '-': '\\u002D',
        ';': '\\u003B',
        u'\u2028': '\\u2028',
        u'\u2029': '\\u2029'
}
# Escape every ASCII character with a value less than 32.
_js_escapes.update(('%c' % z, '\\u%04X' % z) for z in xrange(32))


def escapejs_filter(value):
    escaped = []
    for letter in value:
        if letter in _js_escapes:
            escaped.append(_js_escapes[letter])
        else:
            escaped.append(letter)

    return Markup("".join(escaped))

9390ec89   Goutte   Initial experimen...
133
134
135
136
137
138
139
140
141
142
143
144
tpl_engine = Environment(loader=FileSystemLoader([get_path('view')]),
                         trim_blocks=True,
                         lstrip_blocks=True)

tpl_engine.globals.update(
    url_for=url_for,
    static=static_global,
)

tpl_engine.filters['markdown'] = markdown_filter
tpl_engine.filters['md'] = markdown_filter
tpl_engine.filters['shuffle'] = shuffle_filter
bde97e4d   Goutte   Add more changes ...
145
tpl_engine.filters['escapejs'] = escapejs_filter
9390ec89   Goutte   Initial experimen...
146
147
148
149
150
151
152
153
154
155
156

tpl_global_vars = {
    'request': request,
    'version': version,
    'config': config,
    'now': datetime.datetime.now(),
}


# HELPERS #####################################################################

57f42bd7   Goutte   Log the abortions.
157
158
159
160
161
def abort(code, message):
    log.error(message)
    abort_flask(code, message)


9390ec89   Goutte   Initial experimen...
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
def render_view(view, context=None):
    """
    A simple helper to render [view] template with [context] vars.
    It automatically adds the global template vars defined above, too.
    It returns a string, usually the HTML contents to display.
    """
    context = {} if context is None else context
    return tpl_engine.get_template(view).render(
        dict(tpl_global_vars.items() + context.items())
    )


# def render_page(page, title="My Page", context=None):
#     """
#     A simple helper to render the md_page.html template with [context] vars &
#     the additional contents of `page/[page].md` in the `md_page` variable.
#     It automagically adds the global template vars defined above, too.
#     It returns a string, usually the HTML contents to display.
#     """
#     if context is None:
#         context = {}
#     context['title'] = title
#     context['md_page'] = ''
#     with file(get_path('page/%s.md' % page)) as f:
#         context['md_page'] = f.read()
#     return tpl_engine.get_template('md_page.html').render(
#         dict(tpl_global_vars.items() + context.items())
#     )

077980eb   Goutte   Improve availabil...
191

bc18b96c   Goutte   Implement first (...
192
193
194
195
196
197
198
def is_list_in_list(needle, haystack):
    for n in needle:
        if n not in haystack:
            return False
    return True


2d2af24b   Goutte   Add a basic orbit...
199
def datetime_from_list(time_list):
0b9821dd   Goutte   Clean up.
200
    """
2fedd73b   Goutte   Initial implement...
201
    Datetimes in retrieved CDFs are stored as lists of numbers,
80352490   Goutte   Multi model suppo...
202
203
    with DayOfYear starting at 0. We want it starting at 1 because it's what
    vendor parsers use, both in python and javascript.
0b9821dd   Goutte   Clean up.
204
    """
2d2af24b   Goutte   Add a basic orbit...
205
206
207
208
209
210
    # Day Of Year starts at 0, but for our datetime parser it starts at 1
    doy = '{:03d}'.format(int(''.join(time_list[4:7])) + 1)
    return datetime.datetime.strptime(
        "%s%s%s" % (''.join(time_list[0:4]), doy, ''.join(time_list[7:])),
        "%Y%j%H%M%S%f"
    )
9390ec89   Goutte   Initial experimen...
211

ce8af118   Goutte   Fix the favicon.
212

927c69c3   Goutte   Make the local ca...
213
214
215
216
217
218
219
220
221
222
223
224
225
226
def get_local_filename(url):
    """
    Build the local cache filename for the distant file
    :param url: string
    :return: string
    """
    from slugify import slugify
    n = len('http://')
    if url.startswith('https'):
        n += 1
    s = url[n:]
    return slugify(s)


180d7d97   Goutte   Refactor heavily.
227
def get_target_config(slug):
2fedd73b   Goutte   Initial implement...
228
    for s in config['targets']:  # dumb
8644387c   Goutte   Use real data.
229
230
        if s['slug'] == slug:
            return s
180d7d97   Goutte   Refactor heavily.
231
    raise Exception("No target found in configuration for '%s'." % slug)
8644387c   Goutte   Use real data.
232
233


180d7d97   Goutte   Refactor heavily.
234
235
236
237
238
def check_target_config(slug):
    get_target_config(slug)


def retrieve_amda_netcdf(orbiter, what, started_at, stopped_at):
8644387c   Goutte   Use real data.
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
    """
    Handles remote querying Myriam's API, downloading, extracting and caching
    the netCDF files.
    :param orbiter: key of the source in the YAML config
    :param what: either 'model' or 'orbit', a key in the config of the source
    :param started_at:
    :param stopped_at:
    :return: a list of local file paths to netCDF (.nc) files
    """

    url = config['amda'].format(
        dataSet=what,
        startTime=started_at.isoformat(),
        stopTime=stopped_at.isoformat()
    )
c50cc9d8   Goutte   Continue fixing.
254
    log.info("Fetching remote gzip files list at '%s'." % url)
b2837a08   Goutte   Add three retries...
255
256
    retries = 0
    success = False
92abc15b   Goutte   Mistrust the API ...
257
    errors = []
b2837a08   Goutte   Add three retries...
258
259
260
261
262
263
264
265
    remote_gzip_files = []
    while not success and retries < 3:
        try:
            response = urllib.urlopen(url)
            remote_gzip_files = json.loads(response.read())
            if not remote_gzip_files:
                raise Exception("Failed to fetch data at '%s'." % url)
            if remote_gzip_files == 'NODATASET':
92abc15b   Goutte   Mistrust the API ...
266
267
268
                raise Exception("API says there's no dataset at '%s'." % url)
            if remote_gzip_files == 'ERROR':
                raise Exception("API returned an error at '%s'." % url)
077980eb   Goutte   Improve availabil...
269
            if remote_gzip_files == ['OUTOFTIME']:  # it happens
80352490   Goutte   Multi model suppo...
270
271
                return []
                # raise Exception("API says it's out of time at '%s'." % url)
b2837a08   Goutte   Add three retries...
272
273
274
            success = True
        except Exception as e:
            log.warn("Failed (%d/3) '%s' : %s" % (retries+1, url, e.message))
92abc15b   Goutte   Mistrust the API ...
275
276
            remote_gzip_files = []
            errors.append(e)
b2837a08   Goutte   Add three retries...
277
278
279
        finally:
            retries += 1
    if not remote_gzip_files:
08abc2d4   Goutte   Remove duplicate ...
280
281
282
283
        abort(400, "Failed to fetch gzip files list for %s at '%s' : %s" %
                   (orbiter, url, errors))
    else:
        remote_gzip_files = list(set(remote_gzip_files))
9bfa6c42   Goutte   More bug hunting.
284
285

    log.debug("Fetched remote gzip files list : %s." % str(remote_gzip_files))
8644387c   Goutte   Use real data.
286

8644387c   Goutte   Use real data.
287
288
    local_gzip_files = []
    for remote_gzip_file in remote_gzip_files:
077980eb   Goutte   Improve availabil...
289
290
291
        # hotfixes to remove when fixed upstream @Myriam
        if remote_gzip_file in ['OUTOFTIME', 'ERROR']:
            continue  # sometimes half the response is okay, the other not
8644387c   Goutte   Use real data.
292
        if remote_gzip_file.endswith('/.gz'):
80352490   Goutte   Multi model suppo...
293
            continue  # this is just a plain bug
8644387c   Goutte   Use real data.
294
        remote_gzip_file = remote_gzip_file.replace('cdpp1', 'cdpp', 1)
077980eb   Goutte   Improve availabil...
295
        ################################################
927c69c3   Goutte   Make the local ca...
296
        filename = get_local_filename(remote_gzip_file)
8644387c   Goutte   Use real data.
297
298
299
        local_gzip_file = get_path("../cache/%s" % filename)
        local_gzip_files.append(local_gzip_file)
        if not isfile(local_gzip_file):
9bfa6c42   Goutte   More bug hunting.
300
            log.debug("Retrieving '%s'..." % local_gzip_file)
8644387c   Goutte   Use real data.
301
            urllib.urlretrieve(remote_gzip_file, local_gzip_file)
9bfa6c42   Goutte   More bug hunting.
302
            log.debug("Retrieved '%s'." % local_gzip_file)
8644387c   Goutte   Use real data.
303
304
305
306
307

    local_netc_files = []
    for local_gzip_file in local_gzip_files:
        local_netc_file = local_gzip_file[0:-3]
        local_netc_files.append(local_netc_file)
9bfa6c42   Goutte   More bug hunting.
308
        log.debug("Unzipping '%s'..." % local_gzip_file)
3c064b17   Goutte   Ignore failures w...
309
310
311
312
313
314
315
316
317
318
319
320
        success = True
        try:
            with gzip.open(local_gzip_file, 'rb') as f:
                file_content = f.read()
                with open(local_netc_file, 'w+b') as g:
                    g.write(file_content)
        except Exception as e:
            success = False
            log.warning("Cannot process gz file '%s' from '%s' : %s" %
                        (local_gzip_file, url, e))
        if success:
            log.debug("Unzipped '%s'." % local_gzip_file)
8644387c   Goutte   Use real data.
321
322
323
324

    return local_netc_files


bde97e4d   Goutte   Add more changes ...
325
326
327
# These two configs are not in the YAML config because adding a new parameter
# will not work as-is, you'll have to edit the netcdf-related code

80352490   Goutte   Multi model suppo...
328
# The available parameters in the generated CSV files.
180d7d97   Goutte   Refactor heavily.
329
330
331
# The order matters. If you change this you also need to change the
# innermost loop of `get_data_for_target`.
# The javascript knows the targets' properties under these names.
bde97e4d   Goutte   Add more changes ...
332
PROPERTIES = ('time', 'vrad', 'vtan', 'vtot', 'btan', 'temp', 'pdyn', 'dens',
6491a1f1   Goutte   Fix up the bugs l...
333
              'angl', 'xhee', 'yhee')
180d7d97   Goutte   Refactor heavily.
334

80352490   Goutte   Multi model suppo...
335
336
# The parameters that the users can handle.
# The slug must be one of the properties above.
bc18b96c   Goutte   Implement first (...
337
PARAMETERS = {
ceeb2f4a   Goutte   Add the target co...
338
339
340
    'pdyn': {
        'slug': 'pdyn',
        'name': 'Dyn. Pressure',
5a6d4498   Goutte   Add a title to ea...
341
342
        'title': 'The dynamic pressure.',
        'units': 'nPa',
bde97e4d   Goutte   Add more changes ...
343
344
        'active': True,
        'position': 10,
ceeb2f4a   Goutte   Add the target co...
345
    },
bde97e4d   Goutte   Add more changes ...
346
347
    'vtot': {
        'slug': 'vtot',
bc18b96c   Goutte   Implement first (...
348
        'name': 'Velocity',
5a6d4498   Goutte   Add a title to ea...
349
350
        'title': 'The velocity of the particles.',
        'units': 'km/s',
bde97e4d   Goutte   Add more changes ...
351
352
        'active': False,
        'position': 20,
bc18b96c   Goutte   Implement first (...
353
    },
bde97e4d   Goutte   Add more changes ...
354
355
    'btan': {
        'slug': 'btan',
80352490   Goutte   Multi model suppo...
356
        'name': 'B Tangential',
5a6d4498   Goutte   Add a title to ea...
357
358
        'title': 'B Tangential.',
        'units': 'nT',
bde97e4d   Goutte   Add more changes ...
359
360
        'active': False,
        'position': 30,
bc18b96c   Goutte   Implement first (...
361
    },
ceeb2f4a   Goutte   Add the target co...
362
363
364
    'temp': {
        'slug': 'temp',
        'name': 'Temperature',
5a6d4498   Goutte   Add a title to ea...
365
366
        'title': 'The absolute temperature.',
        'units': 'K',
bde97e4d   Goutte   Add more changes ...
367
368
        'active': False,
        'position': 40,
ceeb2f4a   Goutte   Add the target co...
369
370
371
372
    },
    'dens': {
        'slug': 'dens',
        'name': 'Density',
5a6d4498   Goutte   Add a title to ea...
373
        'title': 'The density N.',
bde97e4d   Goutte   Add more changes ...
374
375
376
        'units': u'cm⁻³',
        'active': False,
        'position': 50,
ceeb2f4a   Goutte   Add the target co...
377
378
379
380
    },
    'angl': {
        'slug': 'angl',
        'name': 'Angle T-S-E',
5a6d4498   Goutte   Add a title to ea...
381
382
        'title': 'Angle Target-Sun-Earth.',
        'units': 'deg',
bde97e4d   Goutte   Add more changes ...
383
384
        'active': False,
        'position': 60,
ceeb2f4a   Goutte   Add the target co...
385
    },
bc18b96c   Goutte   Implement first (...
386
387
}

180d7d97   Goutte   Refactor heavily.
388
389
390
391
392
393

def get_data_for_target(target_config, started_at, stopped_at):
    """
    :return: dict whose keys are datetime as str, values tuples of data
    """
    log.debug("Grabbing data for '%s'..." % target_config['slug'])
80352490   Goutte   Multi model suppo...
394

8644387c   Goutte   Use real data.
395
    try:
80352490   Goutte   Multi model suppo...
396
        models = target_config['models']
077980eb   Goutte   Improve availabil...
397
398
    except Exception as e:
        abort(500, "Invalid model configuration for '%s' : %s"
180d7d97   Goutte   Refactor heavily.
399
400
              % (target_config['slug'], str(e)))
    try:
80352490   Goutte   Multi model suppo...
401
        orbits = target_config['orbit']['models']
180d7d97   Goutte   Refactor heavily.
402
403
404
    except Exception as e:
        abort(500, "Invalid orbit configuration for '%s' : %s"
              % (target_config['slug'], str(e)))
28ef3790   Goutte   Clean up.
405
406

    # Grab the list of netCDF files from Myriam's API
80352490   Goutte   Multi model suppo...
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
    model_files = []
    orbit_files = []
    for model in models:
        model_files = model_files + retrieve_amda_netcdf(
            target_config['slug'], model['slug'], started_at, stopped_at
        )
    for orbit in orbits:
        orbit_files = orbit_files + retrieve_amda_netcdf(
            target_config['slug'], orbit['slug'], started_at, stopped_at
        )
    # Remove possible duplicates
    model_files = set(model_files)
    orbit_files = set(orbit_files)

    # if not len(model_files):
    #     abort(500, "No model files found for '%s'." % target_config['slug'])
    # if not len(orbit_files):
    #     abort(500, "No orbit files found for '%s'." % target_config['slug'])

    precision = "%Y-%m-%dT%H"  # model and orbits times are only equal-ish
180d7d97   Goutte   Refactor heavily.
427
428
    orbit_data = {}  # keys are datetime as str, values arrays of XY
    for orbit_file in orbit_files:
077980eb   Goutte   Improve availabil...
429
        log.debug("%s: opening orbit NETCDF4 '%s'..." %
180d7d97   Goutte   Refactor heavily.
430
431
                  (target_config['name'], orbit_file))
        cdf_handle = Dataset(orbit_file, "r", format="NETCDF4")
8644387c   Goutte   Use real data.
432
        times = cdf_handle.variables['Time']  # YYYY DOY HH MM SS .ms
6491a1f1   Goutte   Fix up the bugs l...
433
434
        data_hee = cdf_handle.variables['HEE']
        for time, datum_hee in zip(times, data_hee):
8644387c   Goutte   Use real data.
435
436
437
            dtime = datetime_from_list(time)
            if started_at <= dtime <= stopped_at:
                dkey = dtime.strftime(precision)
6491a1f1   Goutte   Fix up the bugs l...
438
                orbit_data[dkey] = datum_hee
a7ef1487   Goutte   More logs !
439
        cdf_handle.close()
180d7d97   Goutte   Refactor heavily.
440

8644387c   Goutte   Use real data.
441
442
443
    all_data = {}  # keys are datetime as str, values tuples of data
    for model_file in model_files:
        # Time, StartTime, StopTime, V, B, N, T, Delta_angle, P_dyn
a7ef1487   Goutte   More logs !
444
        log.debug("%s: opening model NETCDF4 '%s'..." %
180d7d97   Goutte   Refactor heavily.
445
                  (target_config['name'], model_file))
8644387c   Goutte   Use real data.
446
447
448
449
450
451
452
453
        cdf_handle = Dataset(model_file, "r", format="NETCDF4")
        times = cdf_handle.variables['Time']  # YYYY DOY HH MM SS .ms
        data_v = cdf_handle.variables['V']
        data_b = cdf_handle.variables['B']
        data_t = cdf_handle.variables['T']
        data_n = cdf_handle.variables['N']
        data_p = cdf_handle.variables['P_dyn']
        data_d = cdf_handle.variables['Delta_angle']
2fedd73b   Goutte   Initial implement...
454
        log.debug("%s: aggregating data from '%s'..." %
180d7d97   Goutte   Refactor heavily.
455
                  (target_config['name'], model_file))
8380e043   Goutte   Fix an awful bug ...
456
        for time, datum_v, datum_b, datum_t, datum_n, datum_p, datum_d \
8644387c   Goutte   Use real data.
457
458
459
460
461
462
                in zip(times, data_v, data_b, data_t, data_n, data_p, data_d):
            vrad = datum_v[0]
            vtan = datum_v[1]
            dtime = datetime_from_list(time)
            if started_at <= dtime <= stopped_at:
                dkey = dtime.strftime(precision)
6491a1f1   Goutte   Fix up the bugs l...
463
464
                x_hee = None
                y_hee = None
180d7d97   Goutte   Refactor heavily.
465
                if dkey in orbit_data:
6491a1f1   Goutte   Fix up the bugs l...
466
467
                    x_hee = orbit_data[dkey][0]
                    y_hee = orbit_data[dkey][1]
8644387c   Goutte   Use real data.
468
469
470
471
                all_data[dkey] = (
                    dtime.strftime("%Y-%m-%dT%H:%M:%S+00:00"),
                    vrad, vtan, sqrt(vrad * vrad + vtan * vtan),
                    datum_b, datum_t, datum_n, datum_p, datum_d,
6491a1f1   Goutte   Fix up the bugs l...
472
                    x_hee, y_hee
8644387c   Goutte   Use real data.
473
474
475
                )
        cdf_handle.close()

180d7d97   Goutte   Refactor heavily.
476
477
478
479
480
481
482
483
484
485
486
487
488
    return all_data


def generate_csv_contents(target_slug, started_at, stopped_at):
    target_config = get_target_config(target_slug)
    log.debug("Crunching CSV contents for '%s'..." % target_config['name'])
    si = StringIO.StringIO()
    cw = csv_writer(si)
    cw.writerow(PROPERTIES)

    all_data = get_data_for_target(target_config, started_at, stopped_at)

    log.debug("Writing and sorting CSV for '%s'..." % target_config['slug'])
8644387c   Goutte   Use real data.
489
490
    for dkey in sorted(all_data):
        cw.writerow(all_data[dkey])
2d2af24b   Goutte   Add a basic orbit...
491

180d7d97   Goutte   Refactor heavily.
492
    log.info("Generated CSV contents for '%s'." % target_config['slug'])
2d2af24b   Goutte   Add a basic orbit...
493
494
    return si.getvalue()

8644387c   Goutte   Use real data.
495

180d7d97   Goutte   Refactor heavily.
496
497
def generate_csv_file_if_needed(target_slug, started_at, stopped_at):
    filename = "%s_%s_%s.csv" % (target_slug,
c0df94bc   Goutte   Adding more logs.
498
499
500
                                 started_at.strftime(FILE_DATE_FMT),
                                 stopped_at.strftime(FILE_DATE_FMT))
    local_csv_file = get_path("../cache/%s" % filename)
80352490   Goutte   Multi model suppo...
501
502
503
504
505
506
507
508
509
510
511
512
513

    generate = True
    if isfile(local_csv_file):
        # It need to have more than one line to not be empty (headers)
        with open(local_csv_file) as f:
            cnt = 0
            for _ in f:
                cnt += 1
                if cnt > 1:
                    generate = False
                    break

    if generate:
c0df94bc   Goutte   Adding more logs.
514
515
516
        log.info("Generating CSV '%s'..." % local_csv_file)
        try:
            with open(local_csv_file, mode="w+") as f:
180d7d97   Goutte   Refactor heavily.
517
                f.write(generate_csv_contents(target_slug,
c0df94bc   Goutte   Adding more logs.
518
519
520
521
                                              started_at=started_at,
                                              stopped_at=stopped_at))
            log.info("Generation of '%s' done." % filename)
        except Exception as e:
5ede388f   Goutte   Make sure failed ...
522
            if isfile(local_csv_file):
92abc15b   Goutte   Mistrust the API ...
523
                log.warn("Removing failed CSV '%s'..." % local_csv_file)
5ede388f   Goutte   Make sure failed ...
524
                removefile(local_csv_file)
9bfa6c42   Goutte   More bug hunting.
525
            abort(500, "Failed creating CSV '%s' : %s" % (filename, e))
c0df94bc   Goutte   Adding more logs.
526
527


28bb4b28   Goutte   API for the cache...
528
529
def remove_files_created_before(date, in_directory):
    """
077980eb   Goutte   Improve availabil...
530
531
532
    Will throw if something horrible happens.
    Does not remove recursively (could be done with os.walk if needed).
    Does not remove directories either.
28bb4b28   Goutte   API for the cache...
533
    :param date: datetime object
077980eb   Goutte   Improve availabil...
534
    :param in_directory: absolute path to directory
28bb4b28   Goutte   API for the cache...
535
536
537
538
539
540
541
    :return:
    """
    import os
    import time

    secs = time.mktime(date.timetuple())

077980eb   Goutte   Improve availabil...
542
543
    if not os.path.isdir(in_directory):
        raise ValueError("No directory to clean at '%s'.")
28bb4b28   Goutte   API for the cache...
544
545
546
547

    removed_files = []
    for file_name in os.listdir(in_directory):
        file_path = os.path.join(in_directory, file_name)
077980eb   Goutte   Improve availabil...
548
549
550
551
552
        if os.path.isfile(file_path):
            t = os.stat(file_path)
            if t.st_ctime < secs:
                os.remove(file_path)
                removed_files.append(file_path)
28bb4b28   Goutte   API for the cache...
553
554
555
556

    return removed_files


077980eb   Goutte   Improve availabil...
557
558
559
560
561
562
563
564
565
566
567
def get_hit_counter():
    hit_count_path = get_path("../VISITS")

    if isfile(hit_count_path):
        hit_count = int(open(hit_count_path).read())
    else:
        hit_count = 1

    return hit_count


a4a9ef03   Goutte   Cache generated C...
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
def increment_hit_counter():
    hit_count_path = get_path("../VISITS")

    if isfile(hit_count_path):
        hit_count = int(open(hit_count_path).read())
        hit_count += 1
    else:
        hit_count = 1

    hit_counter_file = open(hit_count_path, 'w')
    hit_counter_file.write(str(hit_count))
    hit_counter_file.close()

    return hit_count


077980eb   Goutte   Improve availabil...
584
585
586
tpl_global_vars['visits'] = get_hit_counter()


a4a9ef03   Goutte   Cache generated C...
587
588
589
# ROUTING #####################################################################

@app.route('/favicon.ico')
bde97e4d   Goutte   Add more changes ...
590
def favicon():  # we want it served from the root, not from static/
a4a9ef03   Goutte   Cache generated C...
591
592
593
594
595
596
597
598
599
600
    return send_from_directory(
        join(app.root_path, 'static', 'img'),
        'favicon.ico', mimetype='image/vnd.microsoft.icon'
    )


@app.route("/")
@app.route("/home.html")
@app.route("/index.html")
def home():
077980eb   Goutte   Improve availabil...
601
    increment_hit_counter()
bde97e4d   Goutte   Add more changes ...
602
603
    parameters = PARAMETERS.values()
    parameters.sort(key=lambda x: x['position'])
a4a9ef03   Goutte   Cache generated C...
604
605
    return render_view('home.html.jinja2', {
        'targets': config['targets'],
bde97e4d   Goutte   Add more changes ...
606
        'parameters': parameters,
a4a9ef03   Goutte   Cache generated C...
607
608
609
        'planets': [s for s in config['targets'] if s['type'] == 'planet'],
        'probes':  [s for s in config['targets'] if s['type'] == 'probe'],
        'comets':  [s for s in config['targets'] if s['type'] == 'comet'],
077980eb   Goutte   Improve availabil...
610
        'visits':  get_hit_counter(),
a4a9ef03   Goutte   Cache generated C...
611
612
613
    })


180d7d97   Goutte   Refactor heavily.
614
615
@app.route("/<target>_<started_at>_<stopped_at>.csv")
def download_target_csv(target, started_at, stopped_at):
a4a9ef03   Goutte   Cache generated C...
616
617
618
619
620
    """
    Grab data and orbit data for the specified `target`,
    rearrange it and return it as a CSV file.
    `started_at` and `stopped_at` should be UTC.
    """
180d7d97   Goutte   Refactor heavily.
621
    check_target_config(target)
a4a9ef03   Goutte   Cache generated C...
622
    try:
c0df94bc   Goutte   Adding more logs.
623
        started_at = datetime.datetime.strptime(started_at, FILE_DATE_FMT)
a4a9ef03   Goutte   Cache generated C...
624
625
626
    except:
        abort(400, "Invalid started_at parameter : '%s'." % started_at)
    try:
c0df94bc   Goutte   Adding more logs.
627
        stopped_at = datetime.datetime.strptime(stopped_at, FILE_DATE_FMT)
a4a9ef03   Goutte   Cache generated C...
628
629
630
    except:
        abort(400, "Invalid stopped_at parameter : '%s'." % stopped_at)

180d7d97   Goutte   Refactor heavily.
631
    filename = "%s_%s_%s.csv" % (target,
c0df94bc   Goutte   Adding more logs.
632
633
                                 started_at.strftime(FILE_DATE_FMT),
                                 stopped_at.strftime(FILE_DATE_FMT))
a4a9ef03   Goutte   Cache generated C...
634
    local_csv_file = get_path("../cache/%s" % filename)
180d7d97   Goutte   Refactor heavily.
635
    generate_csv_file_if_needed(target, started_at, stopped_at)
a4a9ef03   Goutte   Cache generated C...
636
637
638
639
640
641
    if not isfile(local_csv_file):
        abort(500, "Could not cache CSV file at '%s'." % local_csv_file)

    return send_from_directory(get_path("../cache/"), filename)


0511eed7   Goutte   Tarball generatio...
642
643
@app.route("/<targets>_<started_at>_<stopped_at>.tar.gz")
def download_targets_tarball(targets, started_at, stopped_at):
b2837a08   Goutte   Add three retries...
644
    """
bc18b96c   Goutte   Implement first (...
645
646
647
    Grab data and orbit data for each of the specified `targets`,
    in their own CSV file, and make a tarball of them.
    `started_at` and `stopped_at` should be UTC strings.
b2837a08   Goutte   Add three retries...
648

2fedd73b   Goutte   Initial implement...
649
    targets: string list of targets' slugs, separated by `-`.
b2837a08   Goutte   Add three retries...
650
    """
2fedd73b   Goutte   Initial implement...
651
    separator = '-'
0511eed7   Goutte   Tarball generatio...
652
653
    targets = targets.split(separator)
    targets.sort()
2fedd73b   Goutte   Initial implement...
654
655
    targets_configs = []
    for target in targets:
b2837a08   Goutte   Add three retries...
656
657
        if not target:
            abort(400, "Invalid targets format : `%s`." % targets)
180d7d97   Goutte   Refactor heavily.
658
        targets_configs.append(get_target_config(target))
2fedd73b   Goutte   Initial implement...
659
    if 0 == len(targets_configs):
b2837a08   Goutte   Add three retries...
660
661
        abort(400, "No valid targets specified. What are you doing?")

57493104   Goutte   Add the time to t...
662
    date_fmt = FILE_DATE_FMT
b2837a08   Goutte   Add three retries...
663
664
665
666
667
668
669
670
    try:
        started_at = datetime.datetime.strptime(started_at, date_fmt)
    except:
        abort(400, "Invalid started_at parameter : '%s'." % started_at)
    try:
        stopped_at = datetime.datetime.strptime(stopped_at, date_fmt)
    except:
        abort(400, "Invalid stopped_at parameter : '%s'." % stopped_at)
0511eed7   Goutte   Tarball generatio...
671
672
    sta = started_at.strftime(date_fmt)
    sto = stopped_at.strftime(date_fmt)
b2837a08   Goutte   Add three retries...
673

0511eed7   Goutte   Tarball generatio...
674
    gzip_filename = "%s_%s_%s.tar.gz" % (separator.join(targets), sta, sto)
2fedd73b   Goutte   Initial implement...
675
676
677
    local_gzip_file = get_path("../cache/%s" % gzip_filename)

    if not isfile(local_gzip_file):
0511eed7   Goutte   Tarball generatio...
678
        log.debug("Creating the CSV files for the tarball...")
2fedd73b   Goutte   Initial implement...
679
        for target_config in targets_configs:
0511eed7   Goutte   Tarball generatio...
680
            filename = "%s_%s_%s.csv" % (target_config['slug'], sta, sto)
2fedd73b   Goutte   Initial implement...
681
682
683
            local_csv_file = get_path("../cache/%s" % filename)
            if not isfile(local_csv_file):
                with open(local_csv_file, mode="w+") as f:
180d7d97   Goutte   Refactor heavily.
684
                    f.write(generate_csv_contents(target_config['slug'],
2fedd73b   Goutte   Initial implement...
685
686
687
                                                  started_at=started_at,
                                                  stopped_at=stopped_at))

0511eed7   Goutte   Tarball generatio...
688
        log.debug("Creating the tarball '%s'..." % local_gzip_file)
2fedd73b   Goutte   Initial implement...
689
690
        with tarfile.open(local_gzip_file, "w:gz") as tar:
            for target_config in targets_configs:
0511eed7   Goutte   Tarball generatio...
691
                filename = "%s_%s_%s.csv" % (target_config['slug'], sta, sto)
2fedd73b   Goutte   Initial implement...
692
693
694
695
                local_csv_file = get_path("../cache/%s" % filename)
                tar.add(local_csv_file, arcname=filename)

    if not isfile(local_gzip_file):
0511eed7   Goutte   Tarball generatio...
696
        abort(500, "No tarball to serve. Looked at '%s'." % local_gzip_file)
2fedd73b   Goutte   Initial implement...
697
698

    return send_from_directory(get_path("../cache/"), gzip_filename)
b2837a08   Goutte   Add three retries...
699

28bb4b28   Goutte   API for the cache...
700

bc18b96c   Goutte   Implement first (...
701
702
703
704
705
706
707
708
709
710
@app.route("/<targets>_<params>_<started_at>_<stopped_at>.nc")
def download_targets_netcdf(targets, params, started_at, stopped_at):
    """
    Grab data and orbit data for the specified `target`,
    rearrange it and return it as a CSV file.
    `started_at` and `stopped_at` should be UTC.

    targets: string list of targets' slugs, separated by `-`.
    params: string list of targets' parameters, separated by `-`.
    """
5a6d4498   Goutte   Add a title to ea...
711
    separator = '-'  # /!\ this char should never be in slugs
bc18b96c   Goutte   Implement first (...
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
    targets = targets.split(separator)
    targets.sort()
    targets_configs = []
    for target in targets:
        if not target:
            abort(400, "Invalid targets format : `%s`." % targets)
        targets_configs.append(get_target_config(target))
    if 0 == len(targets_configs):
        abort(400, "No valid targets specified. What are you doing?")
    params = params.split(separator)
    params.sort()
    if 0 == len(params):
        abort(400, "No valid parameters specified. What are you doing?")
    if not is_list_in_list(params, PARAMETERS.keys()):
        abort(400, "Some parameters are not recognized in '%s'." % str(params))

57493104   Goutte   Add the time to t...
728
    date_fmt = FILE_DATE_FMT
bc18b96c   Goutte   Implement first (...
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
    try:
        started_at = datetime.datetime.strptime(started_at, date_fmt)
    except:
        abort(400, "Invalid started_at parameter : '%s'." % started_at)
    try:
        stopped_at = datetime.datetime.strptime(stopped_at, date_fmt)
    except:
        abort(400, "Invalid stopped_at parameter : '%s'." % stopped_at)
    sta = started_at.strftime(date_fmt)
    sto = stopped_at.strftime(date_fmt)

    nc_filename = "%s_%s_%s_%s.nc" % \
                  (separator.join(targets), separator.join(params), sta, sto)
    nc_path = get_path("../cache/%s" % nc_filename)

    if not isfile(nc_path):
        log.debug("Creating the NetCDF file '%s'..." % nc_filename)
        nc_handle = Dataset(nc_path, "w", format="NETCDF4")
        try:
            nc_handle.description = "TODO"  # todo
            nc_handle.history = "Created " + time.ctime(time.time())
            nc_handle.source = "Transplanet (CDDP)"
            available_params = list(PROPERTIES)
            for target in targets_configs:
                target_slug = target['slug']
                log.debug("Adding group '%s' to the NetCDF..." % target_slug)
                nc_group = nc_handle.createGroup(target_slug)
                data = get_data_for_target(target, started_at, stopped_at)
                dkeys = sorted(data)
ceeb2f4a   Goutte   Add the target co...
758
759
                dimension = 'dim_'+target_slug
                nc_handle.createDimension(dimension, len(dkeys))
57493104   Goutte   Add the time to t...
760
761

                # TIME #
ceeb2f4a   Goutte   Add the target co...
762
                nc_time = nc_group.createVariable('time', 'i8', (dimension,))
57493104   Goutte   Add the time to t...
763
764
765
766
767
768
769
770
771
772
773
774
                nc_time.units = "hours since 1970-01-01 00:00:00"
                nc_time.calendar = "standard"
                times = []
                for dkey in dkeys:
                    time_as_string = data[dkey][0][:-6]  # remove +00:00 tail
                    date = datetime.datetime.strptime(time_as_string, date_fmt)
                    times.append(date2num(
                        date, units=nc_time.units, calendar=nc_time.calendar
                    ))
                nc_time[:] = times

                # SELECTED PARAMETERS #
bc18b96c   Goutte   Implement first (...
775
776
777
778
                nc_vars = []
                indices = []
                for param in params:
                    indices.append(available_params.index(param))
ceeb2f4a   Goutte   Add the target co...
779
                    nc_var = nc_group.createVariable(param, 'f8', (dimension,))
5a6d4498   Goutte   Add a title to ea...
780
                    nc_var.units = PARAMETERS[param]['units']
bc18b96c   Goutte   Implement first (...
781
782
783
784
785
786
787
788
                    nc_vars.append(nc_var)
                for i, nc_var in enumerate(nc_vars):
                    index = indices[i]
                    values = []
                    for dkey in dkeys:
                        dval = data[dkey]
                        values.append(dval[index])
                    nc_var[:] = values
ceeb2f4a   Goutte   Add the target co...
789
790

                # ORBIT #
6491a1f1   Goutte   Fix up the bugs l...
791
                nc_x = nc_group.createVariable('xhee', 'f8', (dimension,))
ceeb2f4a   Goutte   Add the target co...
792
                nc_x.units = 'Au'
6491a1f1   Goutte   Fix up the bugs l...
793
                nc_y = nc_group.createVariable('yhee', 'f8', (dimension,))
ceeb2f4a   Goutte   Add the target co...
794
795
796
                nc_y.units = 'Au'
                values_x = []
                values_y = []
6491a1f1   Goutte   Fix up the bugs l...
797
798
                index_x = available_params.index('xhee')
                index_y = available_params.index('yhee')
ceeb2f4a   Goutte   Add the target co...
799
800
801
802
803
804
805
806
                for dkey in dkeys:
                    dval = data[dkey]
                    values_x.append(dval[index_x])
                    values_y.append(dval[index_y])
                nc_x[:] = values_x
                nc_y[:] = values_y
            log.debug("Writing NetCDF '%s'..." % nc_filename)

bc18b96c   Goutte   Implement first (...
807
        except Exception as e:
57493104   Goutte   Add the time to t...
808
            log.error("Failed to generate NetCDF '%s'." % nc_filename)
bc18b96c   Goutte   Implement first (...
809
810
811
812
813
814
815
            raise e
        finally:
            nc_handle.close()

    if not isfile(nc_path):
        abort(500, "No NetCDF to serve. Looked at '%s'." % nc_path)

bde97e4d   Goutte   Add more changes ...
816
    return send_from_directory(get_path("../cache"), nc_filename)
bc18b96c   Goutte   Implement first (...
817
818


28bb4b28   Goutte   API for the cache...
819
820
# API #########################################################################

d9710a98   Goutte   Rename the cleanu...
821
822
@app.route("/cache/cleanup")
def cache_cleanup():
28bb4b28   Goutte   API for the cache...
823
824
    """
    Removes all files from the cache that are older than roughly one month.
927c69c3   Goutte   Make the local ca...
825
    Note: It also removes the .gitkeep file. Maybe it should not.
28bb4b28   Goutte   API for the cache...
826
827
828
829
    """
    a_month_ago = datetime.datetime.now() - datetime.timedelta(days=32)
    cache_dir = get_path('../cache')
    removed_files = remove_files_created_before(a_month_ago, cache_dir)
d9710a98   Goutte   Rename the cleanu...
830
831
832
    count = len(removed_files)
    return "Cache cleaned! Removed %d old file%s." \
           % (count, 's' if count != 1 else '')
28bb4b28   Goutte   API for the cache...
833
834


b500e561   Goutte   Invert the orbits...
835
836
837
838
@app.route("/cache/warmup")
def cache_warmup():
    """
    Warms up the cache for the current day.
927c69c3   Goutte   Make the local ca...
839
    Linked to SpaceWeather#edit in swapp.ls to get the default time interval.
b500e561   Goutte   Invert the orbits...
840
    """
b500e561   Goutte   Invert the orbits...
841
842
843
844
845
    # relativedelta(years=3)
    # startted_at = datetime.datetime.now() - relativedelta(years=3)
    return "To Do"


bde97e4d   Goutte   Add more changes ...
846
847
848
849
850
851
852
@app.route("/run.log")
def run_log():
    with open(get_path('run.log'), 'r') as f:
        contents = f.read()
    return contents


1754789b   Goutte   Decorate and clea...
853
854
855
856
# DEV TOOLS ###################################################################

# @app.route("/inspect")
# def analyze_cdf():
a4a9ef03   Goutte   Cache generated C...
857
#     """
1754789b   Goutte   Decorate and clea...
858
#     For debug purposes.
a4a9ef03   Goutte   Cache generated C...
859
#     """
1754789b   Goutte   Decorate and clea...
860
861
#     cdf_to_inspect = get_path("../res/dummy.nc")
#     cdf_to_inspect = get_path("../res/dummy_jupiter_coordinates.nc")
a4a9ef03   Goutte   Cache generated C...
862
863
#
#     si = StringIO.StringIO()
1754789b   Goutte   Decorate and clea...
864
865
#     cw = csv.DictWriter(si, fieldnames=['Name', 'Shape', 'Length'])
#     cw.writeheader()
a4a9ef03   Goutte   Cache generated C...
866
#
1754789b   Goutte   Decorate and clea...
867
868
869
870
871
872
873
874
875
876
#     # Time, StartTime, StopTime, V, B, N, T, Delta_angle, P_dyn, QualityFlag
#     cdf_handle = Dataset(cdf_to_inspect, "r", format="NETCDF4")
#     for variable in cdf_handle.variables:
#         v = cdf_handle.variables[variable]
#         cw.writerow({
#             'Name': variable,
#             'Shape': v.shape,
#             'Length': v.size,
#         })
#     cdf_handle.close()
a4a9ef03   Goutte   Cache generated C...
877
878
879
880
#
#     return si.getvalue()


9390ec89   Goutte   Initial experimen...
881
882
883
884
885
886
# MAIN ########################################################################

if __name__ == "__main__":
    # Debug mode on, as the production server does not use this.
    extra_files = [get_path('../config.yml')]
    app.run(debug=True, extra_files=extra_files)