ttu
/
CaReCur
mirror of https://github.com/thomastu/CaReCur


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
            """California county shape files.

https://data.ca.gov/dataset/ca-geographic-boundaries/resource/b0007416-a325-4777-9295-368ea6b710e6
"""
import zipfile

from invoke import run
from loguru import logger

from src.conf import settings

RAW_DIR = settings.DATA_DIR / "raw/geography/"
PROCESSED_DIR = settings.DATA_DIR / "processed/geography/"

url = "https://data.ca.gov/dataset/e212e397-1277-4df3-8c22-40721b095f33/resource/b0007416-a325-4777-9295-368ea6b710e6/download/ca-county-boundaries.zip"

fn = "ca-county-boundaries.zip"

if __name__ == "__main__":
    # Get data
    RAW_DIR.mkdir(exist_ok=True)
    PROCESSED_DIR.mkdir(exist_ok=True)

    fp = RAW_DIR/fn

    # Download data
    cmd = f"curl -L {url} -o {fp}"
    run(cmd)

    # Unzip it!
    with zipfile.ZipFile(fp, "r") as fh:
        fh.extractall(PROCESSED_DIR)