Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/github/models.py
| # Copyright (C) 2017 the Software Heritage developers | # Copyright (C) 2017-2018 the Software Heritage developers | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| from sqlalchemy import Column, Boolean, Integer | from sqlalchemy import Column, Boolean, Integer | ||||
| from swh.lister.core.models import ModelBase | from swh.lister.core.models import IndexingModelBase | ||||
| class GitHubModel(ModelBase): | class GitHubModel(IndexingModelBase): | ||||
| """a GitHub repository""" | """a GitHub repository""" | ||||
| __tablename__ = 'github_repos' | __tablename__ = 'github_repos' | ||||
| uid = Column(Integer, primary_key=True) | uid = Column(Integer, primary_key=True) | ||||
| indexable = Column(Integer, index=True) | indexable = Column(Integer, index=True) | ||||
| fork = Column(Boolean) | fork = Column(Boolean) | ||||
| def __init__(self, *args, **kwargs): | def __init__(self, *args, **kwargs): | ||||
| self.fork = kwargs.pop('fork', False) | self.fork = kwargs.pop('fork', False) | ||||
| super().__init__(*args, **kwargs) | super().__init__(*args, **kwargs) | ||||