import pymysql.cursors
from flask import current_app


def get_db():
    db = pymysql.connect(
        host='localhost',
        user='figo',
        password='scolari',
        database='db_figo',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor)
    return db

def get_news(id):
    db = get_db().cursor()
    if id is None:
        query = 'SELECT * FROM news'
        db.execute(query)
        news = db.fetchall()
    else:
        query = f"SELECT * FROM news WHERE id='{id}'"
        db.execute(query)
        news=db.fetchone()
         
    return news

def create_news(title,text):
    slug = title.replace(" ", "-")
    db = get_db()
    db.cursor().execute(
        'INSERT INTO news (title, slug, text)'
        ' VALUES (%s, %s, %s)',
        (title, slug, text)
    )
    db.commit()
    return

def delete_news(id):
    db = get_db()
    db.cursor().execute('DELETE FROM news WHERE id = %s', (id,))
    db.commit()
    return