Source code for hydro.tutorial

import io
import os
from import Mapping
from zipfile import ZipFile

import requests

from . import _hydro_appdirs

[docs] bottle_uri = ""
[docs] bottle_fname = ""
[docs] def _cache_dir(): path = _hydro_appdirs.user_cache_dir os.makedirs(path, exist_ok=True) return path
[docs] def load_cchdo_bottle_data(): """Downloads some CCHDO data for playing with...""" path = os.path.join(_cache_dir(), bottle_fname) with requests.get(bottle_uri, stream=True) as r: r.raise_for_status() with open(path, "wb") as f: for chunk in r.iter_content(chunk_size=8192): if chunk: f.write(chunk)
[docs] class CCHDOBottleData(Mapping): def __init__(self): self.path = os.path.join(_hydro_appdirs.user_cache_dir, bottle_fname) try: with ZipFile(self.path) as f: self.files = f.namelist() except FileNotFoundError: load_cchdo_bottle_data() with ZipFile(self.path) as f: self.files = f.namelist()
[docs] def __len__(self): return len(self.files)
[docs] def __iter__(self): yield from self.files
[docs] def __getitem__(self, key): if key not in self.files: raise KeyError() with ZipFile(self.path) as f: return io.BytesIO(