from peewee import SqliteDatabase, OperationalError


__all__ = [
    'BaseDatabase',
    ]


class BaseDatabase:
    """The base database class to be used with Peewee.
    """
    def __init__(self, url=None):
        self.url = url
        self.db = SqliteDatabase(None)

    def initialize(self, url=None):
        if url is not None:
            self.url = url
        else:
            raise ValueError
        self.db.init(url)
        self._post_initialization()
        return self.db

    def _post_initialization(self):
        from paper_to_git.models import PaperDoc, PaperFolder
        self.db.connect()
        try:
            self.db.create_tables([PaperDoc, PaperFolder])
        except OperationalError:
            # The database already exists.
            pass

from peewee import SqliteDatabase, OperationalError


__all__ = [
    'BaseDatabase',
    ]


class BaseDatabase:
    """The base database class to be used with Peewee.
    """
    def __init__(self, url=None):
        self.url = url
        self.db = SqliteDatabase(None)

    def initialize(self, url=None):
        if url is not None:
            self.url = url
        else:
            raise ValueError
        self.db.init(url)
        self._post_initialization()
        return self.db

    def _post_initialization(self):
        from paper_to_git.models import PaperDoc, PaperFolder, Sync
        self.db.connect()
        try:
            self.db.create_tables([PaperDoc, PaperFolder, Sync])
        except OperationalError:
            # The database already exists.
            pass

Add model Sync to repo.
Add model Sync to repo.

  from peewee import SqliteDatabase, OperationalError
  
  
  __all__ = [
      'BaseDatabase',
      ]
  
  
  class BaseDatabase:
      """The base database class to be used with Peewee.
      """
      def __init__(self, url=None):
          self.url = url
          self.db = SqliteDatabase(None)
  
      def initialize(self, url=None):
          if url is not None:
              self.url = url
          else:
              raise ValueError
          self.db.init(url)
          self._post_initialization()
          return self.db
  
      def _post_initialization(self):
-         from paper_to_git.models import PaperDoc, PaperFolder
+         from paper_to_git.models import PaperDoc, PaperFolder, Sync
          self.db.connect()
          try:
-             self.db.create_tables([PaperDoc, PaperFolder])
+             self.db.create_tables([PaperDoc, PaperFolder, Sync])
          except OperationalError:
              # The database already exists.
              pass
  
Add model Sync to repo.
## Code Before:
from peewee import SqliteDatabase, OperationalError


__all__ = [
    'BaseDatabase',
    ]


class BaseDatabase:
    """The base database class to be used with Peewee.
    """
    def __init__(self, url=None):
        self.url = url
        self.db = SqliteDatabase(None)

    def initialize(self, url=None):
        if url is not None:
            self.url = url
        else:
            raise ValueError
        self.db.init(url)
        self._post_initialization()
        return self.db

    def _post_initialization(self):
        from paper_to_git.models import PaperDoc, PaperFolder
        self.db.connect()
        try:
            self.db.create_tables([PaperDoc, PaperFolder])
        except OperationalError:
            # The database already exists.
            pass

## Instruction:
Add model Sync to repo.
## Code After:
from peewee import SqliteDatabase, OperationalError


__all__ = [
    'BaseDatabase',
    ]


class BaseDatabase:
    """The base database class to be used with Peewee.
    """
    def __init__(self, url=None):
        self.url = url
        self.db = SqliteDatabase(None)

    def initialize(self, url=None):
        if url is not None:
            self.url = url
        else:
            raise ValueError
        self.db.init(url)
        self._post_initialization()
        return self.db

    def _post_initialization(self):
        from paper_to_git.models import PaperDoc, PaperFolder, Sync
        self.db.connect()
        try:
            self.db.create_tables([PaperDoc, PaperFolder, Sync])
        except OperationalError:
            # The database already exists.
            pass
