Models:
Joseph Gordon
joseph-gordon/
cancerTherapy
0
Downloads: 1
[50a3f7]: / modules / RawDB / RDBManager.py

Download this file

93 lines (74 with data), 3.9 kB 

 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import shutil

from IE import CreateCSV

from IE import boolMapToFreq

import sys

sys.path.insert(0, "/home/skjena/cancerTherapy/modules")

from Status.Status import Status



# GETCSV creates a csv file in the specified path

# @row: a tag, specifies whether a row is a cancer or a token

# @column: a tag, specifies whether a column is a cancer or a token



class RDBManager():

    def __init__(self, dataset):

        self.status = Status("RDBManager")

        self.dataset = dataset

        self.datapath = self.GETDATASETFILE()

        self.scaledpath = self.GETSCALEDDATASETFILE()

        self.dataframe = self.boolToFreq()



    def GETCSV(self, row, column, numRows, numCols, path):

        self.status.message(1, "GETCSV(self, row, column, numRows, numCols, path)")

        CreateCSV.createCSV(row, column, numRows, numCols, path)

        self.status.message(0, "GETCSV(self, row, column, numRows, numCols, path)")



    def boolToFreq(self):

        self.status.message(1, "boolToFreq(self)")

        df = None

        if self.dataset == "fm_mutations_independent":

            df = boolMapToFreq.getBoolFrequencies(self.datapath)

        elif self.dataset == "fm_sample_independent":

            df = boolMapToFreq.getFrequencies(self.datapath)

        else:

            df = boolMapToFreq.getCSVFrequencies(self.datapath)



        self.status.message(0, "boolToFreq(self)")

        return df



    def GETDATASETFILE(self):

        self.status.message(1, "GETDATASETFILE(self)")

        if self.dataset == "fm_mutations_independent":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/data/fm_mutations_independent.tab"



        if self.dataset == "fm_sample_independent":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/data/fm_sample_independent.tab"



        if self.dataset == "top10000LungMel":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/cancerTherapy/modules/RawDB/scripts/top10000LungMel.csv"



        if self.dataset == "top1000LungMel":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/cancerTherapy/modules/RawDB/scripts/top1000LungMel.csv"



        if self.dataset == "top100LungMel":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/cancerTherapy/modules/RawDB/scripts/top100LungMel.csv"



        if self.dataset == "top10LungMel":

            self.status.message(0, "GETDATASETFILE(self)")

            return "/home/skjena/cancerTherapy/modules/RawDB/scripts/top10LungMel.csv"



    def GETSCALEDDATASETFILE(self):

        self.status.message(1, "GETSCALEDDATASETFILE(self)")

        if self.dataset == "fm_mutations_independent":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/fm_mutations_independent.csv"



        if self.dataset == "fm_sample_independent":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/fm_sample_independent.csv"



        if self.dataset == "top10000LungMel":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/top10000LungMel.csv"



        if self.dataset == "top1000LungMel":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/top1000LungMel.csv"



        if self.dataset == "top100LungMel":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/top100LungMel.csv"



        if self.dataset == "top10LungMel":

            self.status.message(0, "GETSCALEDDATASETFILE(self)")

            return "/home/skjena/data/scaled/top10LungMel.csv"



    def deleleDirAndContents(fileToDelete):

        path = "/home/skjena/data/" + fileToDelete + "/"

        shutil.rmtree(path)