def __init__(self, dbtype=''): self.cfg = cfg.Config() if dbtype is 'test': self.db_dir = os.getcwd() self.db_name = 'testdb.sqlite' else: self.db_dir = self.cfg.setting('database', 'path') self.db_name = self.cfg.setting('database', 'db') if not (self.db_dir and self.db_name): print 'Error in database path or name. Check cfg file' sys.exit(1) db_path = os.path.join(self.db_dir, self.db_name) print 'Using:', db_path self.db_conn = sqlite3.connect(db_path) self.db_conn.text_factory = str self.db_curr = self.db_conn.cursor()
def __init__(self, db='', path=''): self.error = '' self.cfg = cfg.Config() if not db: self.db_dir = self.cfg.setting('database', 'path') self.db_name = self.cfg.setting('database', 'db') elif db is 'test': self.db_dir = os.getcwd() self.db_name = 'testdb.sqlite' else: if not path: self.db_dir = self.cfg.setting('database', 'path') else: self.db_dir = path self.db_name = db if not self.db_dir or not (os.path.isdir(self.db_dir)) or not self.db_name: sys.stderr.write("GError in database path or name. Check frankenstein.cfg file\n") sys.exit(1) self.db_path = os.path.join(self.db_dir, self.db_name) print('DBGateway connecting: %s' % self.db_path) self.connect(self.db_path)
import zmq import unicodedata import jinja2 import re import urllib import json import string import shutil import threading import thread import cfg ################################################################################ config = cfg.Config() ################################################################################ # zmq def enum(**enums): return type('Enum', (), enums) Modes = enum(IDLE='mode,idle', PREVIEW='mode,preview', CAPTURE='mode,capture') class CamDaemon: def __init__(self):
import torch.nn as nn import torch.nn.functional as F import helper import torch from torch.autograd import Variable import numpy import sys import copy from torch.utils.data import Dataset, DataLoader, random_split import matplotlib.pyplot as plt import cfg Cfg = cfg.Config() class human3DptsDataset(Dataset): def __init__(self): train, label = helper.extractDataLabel(Cfg.dataPath) self.train = torch.from_numpy( numpy.array(train)).type('torch.DoubleTensor') self.label = torch.from_numpy(numpy.array(label)) def __len__(self): return len(self.train) def __getitem__(self, idx): return self.train[idx], self.label[idx] class Net(nn.Module):
def __init__(self): self.cfg = cfg.Config()
import requestsapi import datastore import os import managedata import cfg import redis import logging import logging.config # Cfg logging.config.fileConfig('./conf-files/logger.conf') config = cfg.Config(os.path.abspath('./conf-files/config.cfg')) dbpath = os.path.abspath(config.getLeveldbPath()) rediscfg = config.getRedis() # Api Class apiRequest = requestsapi.RequestData(config.getGoogleKey(), config.getGoogleCompany(), config.getGoogleDbSize()) # ThreatList (might be config) threatlist = apiRequest.getthreatlists() # Store(s) rediscli = redis.StrictRedis(host=rediscfg['host'], port=rediscfg['port'], db=rediscfg['db']) tstore = datastore.ThreatStore(dbpath, threatlist) googleproc = managedata.ProcessingDataFromGoogle(tstore, threatlist, apiRequest) urlmanager = managedata.UrlHashState(threatlist, tstore, rediscli, apiRequest)
from lib.scandir import scandir from storage import StorageFactory from sdhasher import make_sdhash import huntterp import xml_creator from util.str_utils import unescapeHTMLEntities as unescapeHTML from JSAnalysis import analyse try: import argparse except ImportError: print("lack of argparse support") argparse = None LOCK = multiprocessing.Lock() CFG = cfg.Config() class ParserFactory(object): def new_parser(self): parser = None try: parser = ArgParser() except ImportError: parser = GetOptParser() finally: return parser class ParsedArgs(object): """