From 762ba199cd497b10c4648878d63e2c26f95c6033 Mon Sep 17 00:00:00 2001 From: dmy Date: Wed, 19 Feb 2020 17:13:37 +0800 Subject: [PATCH] =?UTF-8?q?1.=E8=A1=A5=E5=85=85=E5=87=A0=E4=B8=AA=E4=B8=BA?= =?UTF-8?q?=E6=9C=80=E7=BB=88=E7=9A=84=E6=96=87=E4=BB=B6=E8=BF=9B=E5=85=A5?= =?UTF-8?q?=E7=89=88=E6=9C=AC=E6=8E=A7=E5=88=B6=E7=B3=BB=E7=BB=9F=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/server/config.py | 8 ++++++++ src/server/db/__init__.py | 5 +++++ src/server/db/connector.py | 5 +++++ src/server/db/model.py | 30 ++++++++++++++++++++++++++++++ src/server/util.py | 14 ++++++++++++++ 5 files changed, 62 insertions(+) create mode 100644 src/server/config.py create mode 100644 src/server/db/__init__.py create mode 100644 src/server/db/connector.py create mode 100644 src/server/db/model.py create mode 100644 src/server/util.py 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) +