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())