CAD_Sketcher/__init__.py at main · davemednick/CAD_Sketcher · 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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
bl_info = {
    "name": "CAD Sketcher",
    "author": "hlorus",
    "version": (0, 20, 0),
    "blender": (2, 80, 0),
    "location": "View3D > Toolbar",
    "description": "Parametric, constraint-based geometry sketcher",
    "warning": "Experimental",
    "category": "3D View",
    "doc_url": "https://hlorus.github.io/CAD_Sketcher",
    "tracker_url": "https://github.com/hlorus/CAD_Sketcher/issues/new/choose",
}

if "bpy" in locals():
    import importlib

    my_modules = (
        theme,
        preferences,
        functions,
        global_data,
        gizmos,
        operators,
        workspacetools,
        class_defines,
        ui,
        install,
        icon_manager,
    )
    for m in my_modules:
        importlib.reload(m)
else:
    import bpy
    from . import (
        preferences,
        functions,
        global_data,
        gizmos,
        operators,
        workspacetools,
        class_defines,
        ui,
        install,
        theme,
        icon_manager,
    )

from tempfile import gettempdir
from pathlib import Path

import logging

logger = logging.getLogger(__name__)

# Clear handlers
if logger.hasHandlers():
    logger.handlers.clear()

logger.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(name)s:{%(levelname)s}: %(message)s")

stream_handler = logging.StreamHandler()
stream_handler.setFormatter(formatter)
logger.addHandler(stream_handler)

filepath = Path(gettempdir()) / (__name__ + ".log")

logger.info("Logging into: " + str(filepath))
file_handler = logging.FileHandler(filepath, mode="w")
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)


def update_logger():
    prefs = functions.get_prefs()
    logger.setLevel(prefs.logging_level)


def ensure_addon_presets(force_write=False):
    import os
    import shutil
    import sys

    scripts_folder = bpy.utils.user_resource("SCRIPTS")
    presets_dir = os.path.join(scripts_folder, "presets", "bgs")

    is_existing = True
    if not os.path.isdir(presets_dir):
        os.makedirs(presets_dir)
        is_existing = False

    if force_write or not is_existing:
        bundled_presets = os.path.join(os.path.dirname(__file__), "presets")
        files = os.listdir(bundled_presets)

        kwargs = {}
        if sys.version_info >= (3, 8):
            kwargs = {"dirs_exist_ok": True}

        shutil.copytree(bundled_presets, presets_dir, **kwargs)

        logger.info("Copy addon presets to: " + presets_dir)


def register():
    # Register base
    ensure_addon_presets()
    theme.register()
    preferences.register()
    install.register()
    update_logger()
    icon_manager.load()

    logger.info(
        "Enabled CAD Sketcher base, version: {}".format(bl_info["version"])
    )

    # Check Module and register all modules
    try:
        install.check_module()
        logger.info("Solvespace available, fully registered modules")
    except ModuleNotFoundError:
        logger.warning(
            "Solvespace module isn't available, only base modules registered"
        )


def unregister():
    install.unregister()
    preferences.unregister()
    theme.unregister()

    if not global_data.registered:
        return

    install.unregister_full()