Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
synergy
o3skim
Commits
39431e78
Commit
39431e78
authored
Sep 21, 2020
by
BorjaEst
Browse files
Add log text at utils module
parent
2dc2df7b
Changes
1
Hide whitespace changes
Inline
Side-by-side
o3skim/utils.py
View file @
39431e78
...
...
@@ -4,6 +4,9 @@ import os
import
yaml
import
netCDF4
import
xarray
as
xr
import
logging
logger
=
logging
.
getLogger
(
'o3skim.utils'
)
@
contextmanager
...
...
@@ -12,15 +15,18 @@ def cd(newdir):
prevdir
=
os
.
getcwd
()
os
.
chdir
(
os
.
path
.
expanduser
(
newdir
))
try
:
logger
.
debug
(
"Temp dir change to: '%s'"
,
newdir
)
yield
finally
:
os
.
chdir
(
prevdir
)
logger
.
debug
(
"Restore directory: '%s'"
,
prevdir
)
def
load
(
yaml_file
):
"""Loads the .yaml file with the sources configurations"""
with
open
(
yaml_file
,
"r"
)
as
ymlfile
:
return
yaml
.
load
(
ymlfile
)
config
=
yaml
.
load
(
ymlfile
)
logging
.
debug
(
"Configuration data: %s"
,
config
)
return
config
def
create_empty_netCDF
(
fname
):
...
...
@@ -34,5 +40,6 @@ def to_netcdf(path, name, dataset):
"""Creates or appends data to named netcdf files"""
years
,
dsx
=
zip
(
*
dataset
.
groupby
(
"time.year"
))
fnames
=
[
path
+
"/"
+
name
+
"_%s.nc"
%
y
for
y
in
years
]
logging
.
info
(
"Save dataset into: %s"
,
fnames
)
[
create_empty_netCDF
(
fn
)
for
fn
in
fnames
if
not
os
.
path
.
isfile
(
fn
)]
xr
.
save_mfdataset
(
dsx
,
fnames
,
mode
=
'a'
)
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment