From 8d4aa91d6f649ce5deb22ba1e13d1b2107d53708 Mon Sep 17 00:00:00 2001
From: mitsuha_s <stmiroslavsckaya@yandex.ru>
Date: Sun, 21 Aug 2022 19:17:44 +0300
Subject: [PATCH] code style changes

---
 database.py                       |  8 +++-----
 migrations/0000.initial-schema.py | 18 ------------------
 migrations/0000.initial_schema.py | 20 ++++++++++++++++++++
 3 files changed, 23 insertions(+), 23 deletions(-)
 delete mode 100644 migrations/0000.initial-schema.py
 create mode 100644 migrations/0000.initial_schema.py

diff --git a/database.py b/database.py
index 9034de3..2dfe1ba 100644
--- a/database.py
+++ b/database.py
@@ -3,10 +3,8 @@ import psycopg2
 from psycopg2.extensions import connection
 from psycopg2.extras import DictCursor, DictRow
 from exceptions import DisplayableException
+from yoyo import read_migrations, get_backend
 from rss import FeedItem
-from yoyo import read_migrations
-from yoyo import get_backend
-
 
 class Database:
     """Implement interaction with the database."""
@@ -17,7 +15,7 @@ class Database:
         self.log.debug('Database.__init__(DSN=\'%s\')', dsn)
         self.conn: connection = psycopg2.connect(dsn)
         self.cur: DictCursor = self.conn.cursor(cursor_factory=DictCursor)
-        self.__migration(dsn)
+        self.__migrate(dsn)
 
     def add_user(self, telegram_id: int) -> int:
         """Add a user's telegram id to the database and return its database id."""
@@ -158,7 +156,7 @@ class Database:
             'INSERT INTO feeds_last_items (feed_id, url, guid) VALUES (%s, %s, %s)', new_items)
         self.conn.commit()
 
-    def __migration(self, dsn: str) -> None:
+    def __migrate(self, dsn: str) -> None:
         backend = get_backend(dsn)
         migrations = read_migrations('./migrations')
 
diff --git a/migrations/0000.initial-schema.py b/migrations/0000.initial-schema.py
deleted file mode 100644
index 1553418..0000000
--- a/migrations/0000.initial-schema.py
+++ /dev/null
@@ -1,18 +0,0 @@
-from yoyo import step
-
-steps = [
-    step('CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, telegram_id INTEGER NOT NULL UNIQUE)'),
-    step('CREATE TABLE IF NOT EXISTS feeds (id SERIAL PRIMARY KEY, url TEXT NOT NULL UNIQUE)'),
-    step('CREATE TABLE IF NOT EXISTS subscriptions ('
-            '   user_id INTEGER REFERENCES users,'
-            '   feed_id INTEGER REFERENCES feeds,'
-            '   UNIQUE (user_id, feed_id)'
-            ')'
-    ),
-    step('CREATE TABLE IF NOT EXISTS feeds_last_items ('
-            '   feed_id INTEGER REFERENCES feeds ON DELETE CASCADE,'
-            '   url TEXT NOT NULL,'
-            '   guid TEXT'
-            ')'
-    )
-]
diff --git a/migrations/0000.initial_schema.py b/migrations/0000.initial_schema.py
new file mode 100644
index 0000000..b81c429
--- /dev/null
+++ b/migrations/0000.initial_schema.py
@@ -0,0 +1,20 @@
+from yoyo import step
+
+steps = [
+    step('CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, telegram_id INTEGER NOT NULL UNIQUE)'),
+    step('CREATE TABLE IF NOT EXISTS feeds (id SERIAL PRIMARY KEY, url TEXT NOT NULL UNIQUE)'),
+    step(
+        'CREATE TABLE IF NOT EXISTS subscriptions ('
+        '   user_id INTEGER REFERENCES users,'
+        '   feed_id INTEGER REFERENCES feeds,'
+        '   UNIQUE (user_id, feed_id)'
+        ')'
+    ),
+    step(
+        'CREATE TABLE IF NOT EXISTS feeds_last_items ('
+        '   feed_id INTEGER REFERENCES feeds ON DELETE CASCADE,'
+        '   url TEXT NOT NULL,'
+        '   guid TEXT'
+        ')'
+    )
+]