Exports config in a new file 'config.py'
This commit is contained in:
parent
1c868d21c0
commit
cff82380b8
28
config.py
Normal file
28
config.py
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
"""
|
||||||
|
Author: freezed <freezed@users.noreply.github.com> 2018-07-27
|
||||||
|
Version: 0.1
|
||||||
|
Licence: `GNU GPL v3` GNU GPL v3: http://www.gnu.org/licenses/
|
||||||
|
|
||||||
|
This file is part of [ocp5](https://github.com/freezed/ocp5) project
|
||||||
|
"""
|
||||||
|
|
||||||
|
# DATABASE
|
||||||
|
DB_CONFIG = {
|
||||||
|
'host': 'localhost',
|
||||||
|
'user': 'loff',
|
||||||
|
'pass': 'loff',
|
||||||
|
'db': 'loff',
|
||||||
|
'char': 'utf8',
|
||||||
|
'file': 'create-db-loff.sql'
|
||||||
|
}
|
||||||
|
|
||||||
|
# API
|
||||||
|
FIELD_KEPT = [
|
||||||
|
'product_name',
|
||||||
|
'stores',
|
||||||
|
'nutrition_grades',
|
||||||
|
'categories_tags'
|
||||||
|
]
|
24
db.py
24
db.py
|
@ -10,21 +10,13 @@ Connect to DB
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import pymysql.cursors
|
import pymysql.cursors
|
||||||
|
from config import DB_CONFIG
|
||||||
|
|
||||||
# CONFIG
|
|
||||||
CONFIG = {
|
|
||||||
'host': 'localhost',
|
|
||||||
'user': 'loff',
|
|
||||||
'pass': 'loff',
|
|
||||||
'db': 'loff',
|
|
||||||
'char': 'utf8',
|
|
||||||
'file': 'create-db-loff.sql'
|
|
||||||
}
|
|
||||||
DB_NOT_FOUND = True
|
DB_NOT_FOUND = True
|
||||||
|
|
||||||
|
|
||||||
# FUNCTION
|
# FUNCTION
|
||||||
def sql_create_db(filename=CONFIG['file']):
|
def sql_create_db(filename=DB_CONFIG['file']):
|
||||||
"""
|
"""
|
||||||
Get the SQL instruction to create the DB for file
|
Get the SQL instruction to create the DB for file
|
||||||
|
|
||||||
|
@ -53,10 +45,10 @@ def sql_create_db(filename=CONFIG['file']):
|
||||||
|
|
||||||
# WORK
|
# WORK
|
||||||
# Connect to the database
|
# Connect to the database
|
||||||
CONNECTION = pymysql.connect(host=CONFIG['host'],
|
CONNECTION = pymysql.connect(host=DB_CONFIG['host'],
|
||||||
user=CONFIG['user'],
|
user=DB_CONFIG['user'],
|
||||||
password=CONFIG['pass'],
|
password=DB_CONFIG['pass'],
|
||||||
charset=CONFIG['char'],
|
charset=DB_CONFIG['char'],
|
||||||
cursorclass=pymysql.cursors.DictCursor)
|
cursorclass=pymysql.cursors.DictCursor)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -67,9 +59,9 @@ try:
|
||||||
|
|
||||||
for idx, val in enumerate(db_list):
|
for idx, val in enumerate(db_list):
|
||||||
|
|
||||||
if CONFIG['db'] in db_list[idx].values():
|
if DB_CONFIG['db'] in db_list[idx].values():
|
||||||
DB_NOT_FOUND = False
|
DB_NOT_FOUND = False
|
||||||
cursor.execute("USE {}".format(CONFIG['db']))
|
cursor.execute("USE {}".format(DB_CONFIG['db']))
|
||||||
print('DB exist : ready to use it.')
|
print('DB exist : ready to use it.')
|
||||||
|
|
||||||
# No DB, create it
|
# No DB, create it
|
||||||
|
|
|
@ -12,14 +12,7 @@ nurition grade.
|
||||||
"""
|
"""
|
||||||
import json
|
import json
|
||||||
import requests
|
import requests
|
||||||
|
from config import FIELD_KEPT
|
||||||
# CONFIG
|
|
||||||
FIELD_KEPT = [
|
|
||||||
'product_name',
|
|
||||||
'stores',
|
|
||||||
'nutrition_grades',
|
|
||||||
'categories_tags'
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def get_product(code):
|
def get_product(code):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user