def __init__(self):
"""
初期化処理
DBと接続する
"""
self.dbManager = dbmanager.DbManager()
def _initdb_(self):
config = configreader.ConfigReader()
config.readconfig()
self._dbman = dbmanager.DbManager(config.username, config.pw
,config.host ,config.db)
self._dbman.connect()
self._dbman.populatedb()
self.tables_col = {'author': 'name', 'title':'title', 'field':'name', 'tag':'name'}
self.tables = {'author': 'author', 'title':'research', 'field':'field', 'tag':'tag'}
self.tables_list = ('author', 'title', 'tag', 'field')
def test_get_feed_list(self):
db = dbmanager.DbManager()
db.execute_query('DELETE FROM feeds;')
db.execute_query(
'INSERT INTO feeds (id, user_id, site_title, site_url, feed_url) VALUES (1, 2, \'site_title\', \'site_url\', \'feed_url\');'
)
feeds = db.get_feed_list()
assert (feeds[0]['id'] == 1)
assert (feeds[0]['user_id'] == 2)
assert (feeds[0]['siteTitle'] == 'site_title')
assert (feeds[0]['siteUrl'] == 'site_url')
assert (feeds[0]['feedUrl'] == 'feed_url')
db.execute_query('DELETE FROM feeds;')
def test_search_recent_updated(self):
db = dbmanager.DbManager()
db.execute_query('DELETE FROM entries;')
db.execute_query(
'INSERT INTO entries (user_id, feed_id, entry_title, entry_url, summary, updated) VALUES (3, 3, \'entry_title\', \'entry_url\', \'summary\', \'2020-01-01 09:00:00\');'
)
db.execute_query(
'INSERT INTO entries (user_id, feed_id, entry_title, entry_url, summary, updated) VALUES (3, 3, \'entry_title\', \'entry_url\', \'summary\', \'2020-01-02 03:01:01\');'
)
db.execute_query(
'INSERT INTO entries (user_id, feed_id, entry_title, entry_url, summary, updated) VALUES (3, 3, \'entry_title\', \'entry_url\', \'summary\', \'2019-12-01 23:59:59\');'
)
date = db.search_recent_updated(3, 3)
assert (date == '2020-01-02T03:01:01')
db.execute_query('DELETE FROM entries;')
def test_close_db(self):
db = dbmanager.DbManager()
db.connect_db()
db.close_db()
assert (db.conn.closed != 0)
# -*- coding: utf-8 -*-
"""
ユーザを追加するツール
Example
-------
python3 create_user.py
"""
import os
import hashlib
from getpass import getpass
import dbmanager
dbman = dbmanager.DbManager()
username = input('username: '******'password: '******'sha256', password.encode(), salt, 100000).hex()
query = 'INSERT INTO users '\
'(name, password_digest, salt) '\
'VALUES (%s, %s, %s)'
dbman.execute_query(query, username, digest, salt.hex())