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
|
"""The base.py module contains parent classes for components."""
import numpy as np
import sys
import os.path
import logging
logging.basicConfig(filename='log.txt',
level=logging.DEBUG,
format='%(asctime)s - %(levelname)s - %(message)s')
class Aircraft:
"""This class tracks all sub-components and is fed to the evaluator."""
pass
class Component:
"""Basic component providing coordinates and tools."""
# TODO: define defaults in separate module
def __init__(self):
self.x = np.array([])
self.z = np.array([])
self.material = str()
self.mass = float()
def set_material(self, material):
"""Set the component bulk material."""
self.material = material
def info_print(self, round):
"""Print all the component's coordinates to the terminal."""
name = f' CREATOR DATA FOR {str(self).upper()} '
num_of_dashes = len(name)
print(num_of_dashes * '-')
print(name)
for k, v in self.__dict__.items():
if type(v) != list:
print('{}:\n'.format(k), v)
print(num_of_dashes * '-')
for k, v in self.__dict__.items():
if type(v) == list:
print('{}:\n'.format(k), np.around(v, round))
return None
def info_save(self, save_path, number):
"""Save all the object's coordinates (must be full path)."""
file_name = f'{str(self).lower()}_{number}.txt'
full_path = os.path.join(save_path, file_name)
try:
with open(full_path, 'w') as sys.stdout:
self.info_print(6)
# This line required to reset behavior of sys.stdout
sys.stdout = sys.__stdout__
logging.debug(f'Successfully wrote to file {full_path}')
except IOError:
print(f'Unable to write {file_name} to specified directory.\n',
'Was the full path passed to the function?')
return None
|