diff --git a/src/server/config.py b/src/server/config.py new file mode 100644 index 0000000..41bc816 --- /dev/null +++ b/src/server/config.py @@ -0,0 +1,8 @@ +import configparser +_config = configparser.ConfigParser() + +def set_configure(file_path): + _config.read(file_path) + +def get(section,item): + return _config[section][item] \ No newline at end of file diff --git a/src/server/db/__init__.py b/src/server/db/__init__.py new file mode 100644 index 0000000..9a046b8 --- /dev/null +++ b/src/server/db/__init__.py @@ -0,0 +1,5 @@ +from db.model import Base +from db.connector import engine + +print("initiating db..") +Base.metadata.create_all(engine) diff --git a/src/server/db/connector.py b/src/server/db/connector.py new file mode 100644 index 0000000..7cd878e --- /dev/null +++ b/src/server/db/connector.py @@ -0,0 +1,5 @@ +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker + +engine = create_engine("mysql+pymysql://dmy:abc123+_@127.0.0.1/compare", echo=True) +Session = sessionmaker(bind=engine) diff --git a/src/server/db/model.py b/src/server/db/model.py new file mode 100644 index 0000000..2111093 --- /dev/null +++ b/src/server/db/model.py @@ -0,0 +1,30 @@ +# from flask_sqlalchemy import SQLAlchemy +# from flask import Flask + +# # import sqlalchmy as sql + +# app = Flask(__name__) +# app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:////tmp/test.db" +# db = SQLAlchemy(app) + +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy import Column, Integer, Text, DATETIME, CHAR +from db.connector import engine +from db.connector import Session + +Base = declarative_base() + + +class Pic(Base): + __tablename__ = "pic" + id = Column(Integer, primary_key=True) + md5 = Column(CHAR(length=34)) + path = Column(Text) + time = Column(DATETIME) + + +def add_pic(md5,path,time): + session=Session() + pic=Pic(md5=md5,path=path,time=time) + session.add(pic) + session.commit() \ No newline at end of file diff --git a/src/server/util.py b/src/server/util.py new file mode 100644 index 0000000..06e634a --- /dev/null +++ b/src/server/util.py @@ -0,0 +1,14 @@ +import hashlib +import datetime + + +def md5(binary_obj): + m = hashlib.md5() + m.update(binary_obj) + return m.hexdigest() + + +def utc_now(): + now = datetime.datetime.now() + return now - datetime.timedelta(hours=8) +