Python SQLalchemy CASE使用之分組累加統計

背景

    通過JenkinsAPI將Jenkins數據入庫CMDB， 來完成開發能效數據的統計展示。通過Jenkins項目構建結果進行分組統計， 統計構建健康度： "每個項目構建的SUCCESS/項目總構建數"。

 

一、表結構

CREATE TABLE `tbl_jenkins_build_test` (
`id` int(11) NOT NULL AUTO_INCREMENT COMMENT 'Build ID',
`name` varchar(32) DEFAULT NULL COMMENT '項目名稱 項目名稱',
`result` varchar(32) DEFAULT NULL COMMENT '構建結果',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=3319 DEFAULT CHARSET=utf8 COMMENT='Jenkins構建記錄表';

二、數據

INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (1, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (2, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (3, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (4, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (5, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (6, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (7, 'test-Agent', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (8, 'test-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (9, 'test-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (10, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (11, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (12, 'dev-Master', 'FAILURE');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (13, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (14, 'dev-Master', 'FAILURE');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (15, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (16, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (17, 'dev-Master', 'SUCCESS');
INSERT INTO `devops`.`tbl_jenkins_build_test`(`id`, `name`, `result`) VALUES (18, 'dev-Master', 'SUCCESS');

三、SQL代碼

select name, sum(case when result='SUCCESS' then 1 else 0 end)/count(result) from tbl_jenkins_build_test GROUP BY name

 

四、Python化

 

4.1 數據模型

class TblJenkinsBuild(Base):
    __tablename__ = 'tbl_jenkins_build_test'
    __table_args__ = {'comment': 'Jenkins構建記錄表'}

    id = Column(Integer, primary_key=True, comment='Build ID')
    name = Column(String(32), comment='項目名稱 項目名稱')
    result = Column(String(32), comment='構建結果')

 

4.2 查詢代碼

from app.models.jenkins import TblJenkinsBuild
from sqlalchemy import case, func

db.session.query(
TblJenkinsBuild.name,
func.sum(case(whens=((TblJenkinsBuild.result == 'SUCCESS', 1),), else_=0)) / func.count(TblJenkinsBuild.result)
).group_by(TblJenkinsBuild.name).all()  # 是SUCCESS則+1， 否則+0

 

五、參考資料

1. SQLAlchemy CASE使用

https://docs.sqlalchemy.org/en/14/core/sqlelement.html#sqlalchemy.sql.expression.case

2. CASE WHEN with ORM (SQLalchemy)

https://stackoverflow.com/questions/11258770/case-when-with-orm-sqlalchemy

3. Python sqlalchemy 模塊，case() 實例源碼
http://codingdict.com/sources/py/sqlalchemy/5994.html