在一个类中使用python方法检查另一个类对象中的变量是否相同

Question

我有2个文件，我正在使用python类方法从中解析数据并创建sql INSERT语句。 在我的两个文件中，我都有一个基因名称列表。 第一个文件是20个基因名称的简单列表，面板名称始终相同。 第二个文件包含有关基因的更多信息，每个基因名称将有多个条目。

目前我的代码如下：

class Gene(object):

    def __init__(self, gene):
        self.gene = gene

    def make_gene_sql(self):
        sql = """INSERT INTO gene ("""+"'"+self.gene+"', '"+panel+"')"
        return sql


class Variants(object):

    def __init__(self, gene, run_id, sample, variant):
        self.run_id = run_id
        self.sample = sample
        self.gene = gene
        self.variant = variant

    def make_vsr_sql(self):
        sql = "INSERT INTO variants_sample_run ("+ "'"+self.sample +"', '"+self.gene +"', '"+self.variant +"', '"+self.run_id+"')"
        return sql



gene_file = open(gene_path, 'r+')
for line in gene_file:
    gene_object = Gene(line.strip('\r\n'))
    gene_sql = gene_object.make_gene_sql()
    cursor.execute(gene_sql)


var_file = open(var_path, 'r+')
for line in var_file.readlines()[1:]:
    item = line.split('\t')
    sample = item[0]
    run = item[1]
    gene = item[2]
    variant = item[3]
    variants_object = Variants(gene, run, sample, variant)
    sql = variants_object.make_var_sql()
    cursor.execute(sql)

我的问题是：有没有办法使Variants类继承自Gene类，并创建一个新函数来检查我的variants_object中的基因变量是否在genes_object中？ 即通过基因类传递我的变量，以确保来自我的var_file的基因变量肯定在基因文件中？ 这样的东西（但实际上有效！）？

class Variants(Gene):

    def __init__(self, gene, run_id, sample, variant):
        self.run_id = run_id
        self.sample = sample
        self.gene = gene.gene
        self.variant = variant

    def check_gene(gene):
        if self.gene not in gene.gene:
            print('Gene not correct')

    def make_vsr_sql(self):
        sql = "INSERT INTO variants_sample_run ("+ "'"+self.sample +"', '"+self.gene +"', '"+self.variant +"', '"+self.run_id+"')"
        return sql

我当时以为这也许可以使用从Gene类创建的包含所有基因的列表，但是我不确定这是否是最好的方法？

Answer 1

类Variants（Gene）：

def __init__(self, gene, run_id, sample, variant):
    self.run_id = run_id
    self.sample = sample
    self.gene = gene.gene
    self.variant = variant

def check_gene(gene):
    if self.gene not in gene.gene:
        print('Gene not correct')

Answer 2

我认为继承不能帮助您解决这一问题。 您可以尝试为所有Gene创建一个类变量：

class Gene(object):
    # The class variable that keeps track of all genes
    genes = []    

    def __init__(self, gene):
        Gene.genes.append(gene)
        self.gene = gene

和您的Variants类：

class Variants(Gene):
    def __init__(self, gene, run_id, sample, variant):
        self.run_id = run_id
        self.sample = sample
        self.gene = gene
        self.variant = variant

    def check_gene(gene):
        if self.gene not in Gene.gene:
            print('Gene not correct')