from enums import AssayType


class EnhancersSiteInfo:
    site = "enhancers"
    assayType = AssayType.Enhancer
    histMark = "H3K27ac"
    name = "enhancer-like"


class PromotersSiteInfo:
    site = "promoters"
    assayType = AssayType.Promoter
    histMark = "H3K4me3"
    name = "promoter-like"


class InteractingGeneSiteInfo:
    site = "interacting_gene"
    assayType = AssayType.InteractingGene
    histMark = "InteractingGene"


SiteInfos = {}
SiteInfos[EnhancersSiteInfo.assayType] = EnhancersSiteInfo
SiteInfos[PromotersSiteInfo.assayType] = PromotersSiteInfo
SiteInfos[InteractingGeneSiteInfo.assayType] = InteractingGeneSiteInfo
