pg_db_appoint_func.py 3.1 KB
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# Date: 2020/7/12 16:27
# @Author:bxh
# @file: pg_db_func.py
import psycopg2
from datetime import *

# 获得连接
def GetConnect():
    conn = False
    try:
        conn = psycopg2.connect(database="mmmworthtest",
                                    user="bianxinhua",
                                    password="bianxinhua_123",
                                    host="114.55.200.59",
                                    port="31543")
    except Exception as err:
        print("连接数据库失败,%s" % err)
    return conn

#获取指派任务单的状态:待发布
def get_appoint_task_status_1_from_database():
    db = GetConnect()
    cursor = db.cursor()
    sql = '''select task_status from tasks  where company_id = 366 and task_type = 3 and task_status = 1 and id = {} '''.format(get_appoint_task_id())
    # print(sql)
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()

#获取指派任务单的状态:进行中
def get_appoint_task_status_3_from_database():
    db = GetConnect()
    cursor = db.cursor()
    sql = '''select task_status from tasks  where company_id = 366 and task_type = 3 and task_status = 3 and id = {} '''.format(get_appoint_task_id())
    # print(sql)
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()
#获取指派任务单的状态:待验收
def get_appoint_task_status_4_from_database():
    db = GetConnect()
    cursor = db.cursor()
    sql = '''select task_status from tasks t where t.company_id = 366 and task_type = 3 and t.task_status = 4  and id = {}'''.format(get_appoint_task_id())
    # print(sql)
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()
#获取指派任务单的状态:已完成
def get_appoint_task_status_5_from_database():
    db = GetConnect()
    cursor = db.cursor()
    sql = '''select task_status from tasks t where t.company_id = 366 and task_type = 3 and t.task_status = 5 and id = {}'''.format(get_appoint_task_id())
    # print(sql)
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()

#获取任务单的状态:已关闭
def get_appoint_task_status_6_from_database():
    db = GetConnect()
    cursor = db.cursor()
    sql = '''select task_status from tasks t where t.company_id = 366 and task_type = 3 and t.task_status = 6 and id = {}'''.format(get_appoint_task_id())
    # print(sql)
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()