Skip to content
Snippets Groups Projects
Commit 42ac7a69 authored by Lionel GUEZ's avatar Lionel GUEZ
Browse files

Parse arguments with dateutil

Files for EUREC4A have format `adt_YYYY-MM-DD.mat` while files for
domain PhD-Lax and global domain have format `adt_YYYY-MM-DD.mat`, so we
need something more general than strptime.
parent 9d9945c6
No related branches found
No related tags found
No related merge requests found
...@@ -17,6 +17,7 @@ import f90nml ...@@ -17,6 +17,7 @@ import f90nml
import sys import sys
import pty import pty
import os import os
from dateutil import parser
def define_fields(writers): def define_fields(writers):
"""writers is a dictionary of shapefile.Writer objects.""" """writers is a dictionary of shapefile.Writer objects."""
...@@ -95,10 +96,12 @@ def write(writers, cell, cyclone): ...@@ -95,10 +96,12 @@ def write(writers, cell, cyclone):
if len(sys.argv) == 1: sys.exit("Required argument: first input .mat file") if len(sys.argv) == 1: sys.exit("Required argument: first input .mat file")
adt_file = sys.argv[1] adt_file = sys.argv[1]
dirname, basename = path.split(adt_file) dirname, basename = path.split(adt_file)
my_date = datetime.datetime.strptime(basename, "adt_%Y-%m-%d.mat").date() my_date = parser.parse(basename, yearfirst=True, dayfirst = False,
fuzzy = True).date()
if len(sys.argv) == 3: if len(sys.argv) == 3:
final_date = datetime.datetime.strptime(sys.argv[2], "%Y-%m-%d").date() final_date = parser.parse(sys.argv[2], yearfirst=True, dayfirst = False)\
.date()
else: else:
final_date = my_date final_date = my_date
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment