kraken/backend/settings/config.py

"""Module core/configs provide project base settings"""
import glob
import json
import os
import os.path


class Settings:
    PROJECT_NAME: str = 'Kraken'
    DEVICE = 'neo'
    CONFIG_INFO = {
        'host': '127.0.0.1',
        'port': 8712,
        'channel_count': 9,
        'sample_rate': 1000,
        'buffer_length': 0.5,
        'channel_labels': [
            'CH001',
            'CH002',
            'CH003',
            'CH004',
            'CH005',
            'CH006',
            'CH007',
            'CH008',
            'STIM'
        ],
        'strips': [['CH001', 'CH002', 'CH003', 'CH004'], 
                   ['CH005', 'CH006', 'CH007', 'CH008']],
    }
    FINGERMODEL_IDS = {
        'rest': 0,
        'cylinder': 1,
        'ball': 2,
        'flex': 3,
        'double': 4,
        'treble': 5
    }
    FINGERMODEL_IDS_INVERSE = {
        0: 'rest',
        1: 'cylinder',
        2: 'ball',
        3: 'flex',
        4: 'double',
        5: 'treble'
    }
    PROJECT_VERSION: str = '0.0.1'
    DATA_PATH = './data'

    def __init__(self):
        self.config = {"lang": "zh"}

    def get_message(self):
        self.message = {}
        msg_list = glob.glob('./static/config/message*.json')
        for msg in msg_list:
            filename = os.path.basename(msg)
            msg_code, ext = os.path.splitext(filename)
            with open(msg, 'r', encoding='utf8') as file:
                self.message[msg_code.split('_')[1]] = json.load(file)
        return self.message


settings = Settings()