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):
"""