58 lines
1.2 KiB
Python
58 lines
1.2 KiB
Python
import mariadb
|
|
import sys
|
|
from decouple import config
|
|
|
|
|
|
database_info = {
|
|
'database': config('MDB_DATABASE'),
|
|
'host': config('MDB_HOST'),
|
|
'port': config('MDB_PORT', cast=int),
|
|
'user': config('MDB_USER'),
|
|
'password': config('MDB_PASSWORD')
|
|
}
|
|
|
|
print("connecting to mariadb")
|
|
|
|
# Connect to MariaDB Platform
|
|
try:
|
|
conn = mariadb.connect(**database_info)
|
|
except mariadb.Error as e:
|
|
print(f"Error connecting to MariaDB Platform: {e}")
|
|
sys.exit(1)
|
|
|
|
print("connected to mariadb")
|
|
|
|
|
|
# Get Cursor
|
|
cur = conn.cursor()
|
|
|
|
# class BaseModel(Model):
|
|
# class Meta:
|
|
# database = db
|
|
#
|
|
#
|
|
# class os2022_ngrams(BaseModel):
|
|
# file_id = IntegerField()
|
|
# sent_id = FloatField()
|
|
# ngram_len = IntegerField()
|
|
# frequency_g_t = IntegerField()
|
|
# gram_text = TextField()
|
|
# lemma_text = TextField()
|
|
# xpos_text = TextField()
|
|
# upos_text = TextField()
|
|
#
|
|
# db.connect()
|
|
|
|
|
|
class Ngrams_Manager:
|
|
@staticmethod
|
|
def get_by_file_id(file_id):
|
|
try:
|
|
# cur.execute(f'SELECT * from os2022_ngrams WHERE file_id = {file_id}')
|
|
# cur.execute(f'SELECT COUNT(*) FROM os2022_ngrams')
|
|
# return list(cur)
|
|
return 1
|
|
except Exception as e:
|
|
print(e, 'EXC')
|
|
return 0
|