M-STAR/laplace_tests.py

from load_data import load_DBs
class Mydata:
    def __init__(self, load):
        if load:
            (self.dsatable, self.src2dsa, self.dsa2cve, self.cvetable, self.src2month, self.src2sloccount, self.src2pop, self.src2deps, self.pkg_with_cvss) = load_DBs()
        else:
            print('no load command given')

def main():
    data = Mydata(True)
    print('Data loading done')
    #
    years = 18
    # 2001-2018
    
    i = calc_laplace(data, years)
    i.calculate()

class calc_laplace:
    def __init__(self,data,years):
        self. src2dsa = data.src2dsa
        self.dsa2cve = data.dsa2cve
        self.cvetable = data.cvetable
        self.pkg_with_cvss = data.pkg_with_cvss
        self.95percent = 1.96
        self.90percent = 1.64

    def laplace_wheezy(self):
        pass

    def laplace_package(self, pkg):
        pass

if __name__ == "__main__":
    main()