Source code for bw2data.backends.schema

from peewee import DoesNotExist, TextField

from bw2data.errors import UnknownObject
from bw2data.snowflake_ids import SnowflakeIDBaseClass
from bw2data.sqlite import PickleField




class ActivityDataset(SnowflakeIDBaseClass):


    data = PickleField()  # Canonical, except for other C fields



    code = TextField()  # Canonical



    database = TextField()  # Canonical



    location = TextField(null=True)  # Reset from `data`



    name = TextField(null=True)  # Reset from `data`



    product = TextField(null=True)  # Reset from `data`



    type = TextField(null=True)  # Reset from `data`


    @property


    def key(self):
        return (self.database, self.code)






class ExchangeDataset(SnowflakeIDBaseClass):


    data = PickleField()  # Canonical, except for other C fields



    input_code = TextField()  # Canonical



    input_database = TextField()  # Canonical



    output_code = TextField()  # Canonical



    output_database = TextField()  # Canonical



    type = TextField()  # Reset from `data`






def get_id(key):
    if isinstance(key, int):
        try:
            ActivityDataset.get(ActivityDataset.id == key)
        except DoesNotExist:
            raise UnknownObject
        return key
    else:
        try:
            return ActivityDataset.get(
                ActivityDataset.database == key[0], ActivityDataset.code == key[1]
            ).id
        except DoesNotExist:
            raise UnknownObject