from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker import re from datetime import date, datetime from model.pls import PLS from model.plw import PLW from model.sd import SD from model.klb import KLB from model.football_sfc import FootballSfc from model.football_match import FootballMatch from model.football_team import FootballTeam from model.football_match_bf_odds import FootballMatchBfOdd from model.football_match_bqc_odds import FootballMatchBqcOdd from model.football_match_spf_odds import FootballMatchSpfOdd from model.football_match_zjq_odds import FootballMatchZjqOdd class Lottery(object): """ Lottery Base Object """ def __init__(self, lottery_type='pls'): self._pat1 = re.compile('开奖日期：(\d+)年(\d+)月(\d+)日.*') self._pat2 = re.compile('开奖日期：(\d+)-(\d+)-(\d+)\s.*') self._lottery_type = lottery_type self.db = self._get_db_session() if lottery_type.lower() == 'pls': self._Model = PLS elif lottery_type.lower() == 'sd': self._Model = SD elif lottery_type.lower() == 'plw': self._Model = PLW elif lottery_type.lower() == 'sfc': self._Model = FootballSfc elif lottery_type.lower() == 'klb': self._Model = KLB else: raise Exception("未知的lottery_type") def _get_db_session(self): _engine = create_engine("mysql+pymysql://test:123456@localhost/lottery?charset=utf8", pool_pre_ping=True, pool_recycle=3600) _DbSession = sessionmaker(bind=_engine) return _DbSession() def __enter__(self): return self def __exit__(self, type, value, trace): print("close the db session...") self.db.close() def get_last_id_by_sum_num(self, sum_num): """ 根据总和获取id """ last_id = self.db.query(self._Model.id).filter_by(sum_num=sum_num).order_by(self._Model.id.desc()).first() if last_id is not None: return last_id[0] return None def insert(self, draw_issue, draw_date, draw_code, **kwargs): """ 插入3d、排列三、排列五 """ result = self.db.query(self._Model).filter_by(draw_issue=draw_issue).first() if result is None: record = self._Model() record.draw_issue = draw_issue record.draw_code = draw_code record.created_at = datetime.now() m1 = re.match(self._pat1, draw_date) m2 = re.match(self._pat2, draw_date) if m1 or m2 : if m1: record.draw_date = date(int(m1.group(1)), int(m1.group(2)), int(m1.group(3))) else: record.draw_date = date(int(m2.group(1)), int(m2.group(2)), int(m2.group(3))) else: raise Exception(f"issue:{draw_issue}数据写入失败。。。") # 如果是排列五、排列五和3D if isinstance(record, (PLS, SD, PLW)): record.sum_num = sum(map(int, draw_code)) record.code_small = len(list(filter(lambda x: True if int(x) < 5 else False, draw_code))) record.code_big = len(list(filter(lambda x: True if int(x) >= 5 else False, draw_code))) record.code_single = len(list(filter(lambda x: True if int(x) % 2 == 1 else False, draw_code))) record.code_double = len(list(filter(lambda x: True if int(x) % 2 == 0 else False, draw_code))) if isinstance(record, (PLS, SD)): record.hundred = draw_code[0] record.ten = draw_code[1] record.one = draw_code[2] record.sum_hundred_one = int(draw_code[2]) + int(draw_code[0]) record.sum_hundred_ten = int(draw_code[2]) + int(record.draw_code[1]) record.sum_ten_one = int(record.draw_code[1]) + int(record.draw_code[0]) if len(set(draw_code)) == 2: record.group_type = 3 elif len(set(draw_code)) == 3: record.group_type = 6 else: record.group_type = 1 elif isinstance(record, PLW): record.ten_thousand = draw_code[0] record.thousand = draw_code[1] record.hundred = draw_code[2] record.ten = draw_code[3] record.one = draw_code[4] record.sum_hundred_one = int(draw_code[2]) + int(draw_code[4]) record.sum_hundred_ten = int(draw_code[2]) + int(record.draw_code[3]) record.sum_ten_one = int(record.draw_code[3]) + int(record.draw_code[4]) elif isinstance(record, FootballSfc): record.open_result = draw_code record.sale_money = kwargs["sfc_sale"] record.r9_sale_money = kwargs["rj_sale"] record.award_1_money = kwargs["award_1_money"] record.award_1_num = kwargs["award_1_num"] record.award_2_money = kwargs["award_2_money"] record.award_2_num = kwargs["award_2_num"] record.r9_award_money = kwargs["r9_award_money"] record.r9_award_num = kwargs["r9_award_num"] self.db.add(record) self.db.commit() return record.id else: print(f"{draw_issue}开奖结果已经写入数据库中...") return None def insert_match(self, match_id, **kwargs): """ 插入比赛结果数据 """ result = self.db.query(FootballMatch).filter_by(match_id=match_id).first() if result is not None: print(f"{match_id}已经写入数据库中。。。") return team1 = self.db.query(FootballTeam).filter_by(team_id=kwargs["homeTeamId"]).first() if team1 is None: home_team = FootballTeam(team_id=kwargs["homeTeamId"], team_name=kwargs["homeTeam"], team_all_name=kwargs["allHomeTeam"]) self.db.add(home_team) self.db.commit() team2 = self.db.query(FootballTeam).filter_by(team_id=kwargs["awayTeamId"]).first() if team2 is None: away_team = FootballTeam(team_id=kwargs["awayTeamId"], team_name=kwargs["awayTeam"], team_all_name=kwargs["allAwayTeam"]) self.db.add(away_team) self.db.commit() football_match = FootballMatch(match_id=match_id) football_match.match_num = kwargs["matchNum"] football_match.match_num_str = kwargs["matchNumStr"] football_match.match_date = kwargs["matchDate"] football_match.league_id = kwargs["leagueId"] football_match.league_name = kwargs["leagueNameAbbr"] football_match.home_team_id = kwargs["homeTeamId"] football_match.home_name = kwargs["homeTeam"] football_match.away_team_id = kwargs["awayTeamId"] football_match.away_name = kwargs["awayTeam"] football_match.half_score = kwargs["sectionsNo1"] football_match.score = kwargs["sectionsNo999"] football_match.is_single = kwargs["bettingSingle"] football_match.match_result_status = kwargs["matchResultStatus"] if int(kwargs["matchResultStatus"]) != 0: # 0 代表比赛取消 if kwargs["winFlag"] == "A": football_match.result = 0 elif kwargs["winFlag"] == "D": football_match.result = 1 elif kwargs["winFlag"] == "H": football_match.result = 3 else: result = None if kwargs["goalLine"] is not None: football_match.rq_count = int(kwargs["goalLine"]) else: football_match.rq_count = None if ":" in kwargs["sectionsNo1"]: half_score = list(map(int, kwargs["sectionsNo1"].split(":"))) if half_score[0] < half_score[1]: football_match.bqc_result = f"0{football_match.result}" elif half_score[0] == half_score[1]: football_match.bqc_result = f"1{football_match.result}" else: football_match.bqc_result = f"3{football_match.result}" if ":" in kwargs["sectionsNo999"]: scores = list(map(int, kwargs["sectionsNo999"].split(":"))) football_match.total_goal_count = sum(scores) if scores[0] + int(kwargs["goalLine"]) < scores[1]: football_match.rq_result = 0 elif scores[0] + int(kwargs["goalLine"]) == scores[1]: football_match.rq_result = 1 else: football_match.rq_result = 3 else: football_match.total_goal_count = None football_match.rq_result = None self.db.add(football_match) self.db.commit() print(f"{football_match.id}:{football_match.match_id}写入数据库完成。。。")