setup_project/setup_project.py at master · blt2589/setup_project · GitHub

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
setup_project.py -- GIS Project Setup Utility
Garin Wally; May 2014/May 2016

This script creates a project folder-environment for GIS projects as follows:
<project_name>/
    data/
        raw/
        <project_name>.gdb
    design/
        fonts/
        images/
        rasters/
        vectors/
    maps/
        archive/
        final/
    reports/
    resources/
        tools/
"""

import argparse
import os
from shutil import copy2

# import arcpy  # moved for speed


# =============================================================================
# CLI ARGS

argp = argparse.ArgumentParser(description=__doc__, add_help=True,
                               formatter_class=argparse.RawTextHelpFormatter)
argp.add_argument("-n", action="store", dest="name",
                  help="Name of new project folder.")
argp.add_argument("--gdb", action="store_true", default=False,
                  dest="make_gdb", help="Option to make a gdb on setup.")
argp.add_argument("--cart", action="store_true", default=False,
                  dest="make_cart",
                  help="Option to make cartographic resource folders.")
# TODO: For now, we assume user has $ cd'd into the desired base directory

args = argp.parse_args()

if args.make_gdb:
    import arcpy

# =============================================================================
# FUNCTIONS


def make_gdb(gdb_name):
    # if no project exists, place gdb in cwd
    # if it does, place in DATA/
    # process: make gdb in C:/temp and copy to cwd
    copy2(None, None)
    pass


def main(dest_folder):
    # <project>/
    os.mkdir(dest_folder)
    # DATA/
    #   RAWDATA/
    os.makedirs(os.path.join(dest_folder, "data", "raw"))
    #       TODO: make scratch gdb here? / use C:/temp?
    #   MISC/
    os.mkdir(os.path.join(dest_folder, "data", "misc"))
    #   GDB/
    # TODO: if args.make_gdb: make_gdb(dest_folder)

    # MAPS/
    #   archive/
    os.makedirs(os.path.join(dest_folder, "maps", "archive"))
    #   FINAL/
    os.mkdir(os.path.join(dest_folder, "maps", "final"))

    if args.make_cart:
        # DESIGN/
        #   FONTS/
        os.makedirs(os.path.join(dest_folder, "design", "fonts"))
        #   IMAGES/
        os.mkdir(os.path.join(dest_folder, "design", "images"))
        #   RASTERS/
        os.mkdir(os.path.join(dest_folder, "design", "rasters"))
        #   VECTORS/
        os.mkdir(os.path.join(dest_folder, "design", "vectors"))

    # REPORTS/
    os.mkdir(os.path.join(dest_folder, "reports"))

    # RESOURCES/
    #   TOOLS/
    os.makedirs(os.path.join(dest_folder, "resources", "tools"))

    return


# =============================================================================
# RUN IT

if __name__ == "__main__":
    if not args.name:
        args.name = "new_project"

    new_proj = os.path.join(os.getcwd(), args.name)

    main(new_proj)