# encoding: utf-8 import sys import io from common.logging import Logging COUNTRY_REVERSE = {"安道尔共和国": "AD", "阿拉伯联合酋长国": "AE", "阿富汗": "AF", "安提瓜和巴布达": "AG", "安圭拉岛": "AI", "阿尔巴尼亚": "AL", "亚美尼亚": "AM", "安哥拉": "AO", "阿根廷": "AR", "奥地利": "AT", "澳大利亚": "AU", "阿塞拜疆": "AZ", "巴巴多斯": "BB", "孟加拉国": "BD", "比利时": "BE", "布基纳法索": "BF", "保加利亚": "BG", "巴林": "BH", "布隆迪": "BI", "贝宁": "BJ", "巴勒斯坦": "BL", "百慕大群岛": "BM", "文莱": "BN", "玻利维亚": "BO", "巴西": "BR", "巴哈马": "BS", "博茨瓦纳": "BW", "白俄罗斯": "BY", "伯利兹": "BZ", "加拿大": "CA", "中非共和国": "CF", "刚果": "CG", "瑞士": "CH", "库克群岛": "CK", "智利": "CL", "喀麦隆": "CM", "中国": "CN", "哥伦比亚": "CO", "哥斯达黎加": "CR", "古巴": "CU", "塞浦路斯": "CY", "捷克": "CZ", "德国": "DE", "吉布提": "DJ", "丹麦": "DK", "多米尼加共和国": "DO", "阿尔及利亚": "DZ", "厄瓜多尔": "EC", "爱沙尼亚": "EE", "埃及": "EG", "西班牙": "ES", "埃塞俄比亚": "ET", "芬兰": "FI", "斐济": "FJ", "法国": "FR", "加蓬": "GA", "英国": "GB", "格林纳达": "GD", "格鲁吉亚": "GE", "法属圭亚那": "GF", "加纳": "GH", "直布罗陀": "GI", "冈比亚": "GM", "几内亚": "GN", "希腊": "GR", "危地马拉": "GT", "关岛": "GU", "圭亚那": "GY", "香港特别行政区": "HK", "洪都拉斯": "HN", "海地": "HT", "匈牙利": "HU", "印度尼西亚": "ID", "爱尔兰": "IE", "以色列": "IL", "印度": "IN", "伊拉克": "IQ", "伊朗": "IR", "冰岛": "IS", "意大利": "IT", "牙买加": "JM", "约旦": "JO", "日本": "JP", "肯尼亚": "KE", "吉尔吉斯坦": "KG", "柬埔寨": "KH", "朝鲜": "KP", "韩国": "KR", "科特迪瓦共和国": "KT", "科威特": "KW", "哈萨克斯坦": "KZ", "老挝": "LA", "黎巴嫩": "LB", "圣卢西亚": "LC", "列支敦士登": "LI", "斯里兰卡": "LK", "利比里亚": "LR", "莱索托": "LS", "立陶宛": "LT", "卢森堡": "LU", "拉脱维亚": "LV", "利比亚": "LY", "摩洛哥": "MA", "摩纳哥": "MC", "摩尔多瓦": "MD", "马达加斯加": "MG", "马里": "ML", "缅甸": "MM", "蒙古": "MN", "澳门": "MO", "蒙特塞拉特岛": "MS", "马耳他": "MT", "毛里求斯": "MU", "马尔代夫": "MV", "马拉维": "MW", "墨西哥": "MX", "马来西亚": "MY", "莫桑比克": "MZ", "纳米比亚": "NA", "尼日尔": "NE", "尼日利亚": "NG", "尼加拉瓜": "NI", "荷兰": "NL", "挪威": "NO", "尼泊尔": "NP", "瑙鲁": "NR", "新西兰": "NZ", "阿曼": "OM", "巴拿马": "PA", "秘鲁": "PE", "巴布亚新几内亚": "PG", "菲律宾": "PH", "巴基斯坦": "PK", "波兰": "PL", "波多黎各": "PR", "葡萄牙": "PT", "巴拉圭": "PY", "卡塔尔": "QA", "罗马尼亚": "RO", "俄罗斯": "RU", "沙特阿拉伯": "SA", "所罗门群岛": "SB", "塞舌尔": "SC", "苏丹": "SD", "瑞典": "SE", "新加坡": "SG", "斯洛文尼亚": "SI", "斯洛伐克": "SK", "塞拉利昂": "SL", "圣马力诺": "SM", "塞内加尔": "SN", "索马里": "SO", "苏里南": "SR", "圣多美和普林西比": "ST", "萨尔瓦多": "SV", "叙利亚": "SY", "斯威士兰": "SZ", "乍得": "TD", "多哥": "TG", "泰国": "TH", "塔吉克斯坦": "TJ", "土库曼斯坦": "TM", "突尼斯": "TN", "汤加": "TO", "土耳其": "TR", "特立尼达和多巴哥": "TT", "台湾": "TW", "坦桑尼亚": "TZ", "乌克兰": "UA", "乌干达": "UG", "美国": "US", "乌拉圭": "UY", "乌兹别克斯坦": "UZ", "圣文森特岛": "VC", "委内瑞拉": "VE", "越南": "VN", "也门": "YE", "南斯拉夫": "YU", "南非": "ZA", "赞比亚": "ZM", "扎伊尔": "ZR", "津巴布韦": "ZW", } def shorten(floors): floors.sort(key=lambda x: x) # print floors shortened = [] cur_index = 0 while cur_index < len(floors): next_index = cur_index + 1 while next_index < len(floors): if floors[cur_index] * 1.14 < floors[next_index]: break next_index += 1 shortened.append(floors[next_index - 1]) cur_index = next_index return shortened def find_match_floor(floor, floors): for f in floors: if f >= floor: return float(f) / 100 def combine_floors(platform, file_path, write_to_path): file_handler = open(file_path, 'r') floor_data = file_handler.read() floors = floor_data.split('\n') floor0 = [] floor1 = [] floor2 = [] floor3 = [] for i in range(0, len(floors) / 2): cur_floors = floors[2 * i + 1].split(',') if len(cur_floors) <= 5: continue cur_floors = [int(f.strip()) for f in cur_floors] if cur_floors[2] not in floor0: floor0.append(cur_floors[2]) if cur_floors[4] not in floor1: floor1.append(cur_floors[4]) if cur_floors[6] not in floor2: floor2.append(cur_floors[6]) if cur_floors[7] not in floor3: floor3.append(cur_floors[7]) floor0 = shorten(floor0) floor1 = shorten(floor1) floor2 = shorten(floor2) floor3 = shorten(floor3) result_str = 'countryCode,platform,placementId,targetPrice\n' for i in range(0, len(floors) / 2): country_code = COUNTRY_REVERSE[floors[i * 2]] if len(floors[2 * i + 1]) == 0: continue cur_floors = floors[2 * i + 1].split(',') cur_floors = [int(f) for f in cur_floors] result_str += '%s,%s,unity_inter_I1_lv1,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[1], floor0)) result_str += '%s,%s,unity_inter_I1_lv2,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[3], floor1)) result_str += '%s,%s,unity_inter_I2_lv1,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[5], floor2)) result_str += '%s,%s,unity_inter_I2_lv2,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[7], floor3)) # result_str += '%s,%s,inter_I1_lv1,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[2], floor0) * 1.7) # result_str += '%s,%s,inter_I1_lv2,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[4], floor1) * 1.7) # result_str += '%s,%s,inter_I2_lv1,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[6], floor2) * 1.7) # result_str += '%s,%s,inter_I2_lv2,%.2f\n' % (country_code, platform, find_match_floor(cur_floors[7], floor3) * 1.7) open(write_to_path, "w").write(result_str) print "文件写入完成: {}".format(write_to_path) def _printScriptInfo(): print "************************************************************" print "* 脚本作用： 生成unity底价csv" print "* 参数说明：" print "* 游戏所在平台 (android | ios)" print "* 插屏价格配置文件路径" print "* 更新后写入的文件" print "* 例：" print "* bulldog price_script_unity inter.txt unity.csv" print "************************************************************" print "" def run(argv): print argv _printScriptInfo() if len(argv) < 4: Logging.error("参数个数小于3个") exit(0) platform = argv[1] txt_path = argv[2] write_to_path = argv[3] Logging.info("配置文件路径: {}".format(txt_path)) Logging.info("写入文件路径: {}".format(write_to_path)) combine_floors(platform, txt_path, write_to_path) if __name__ == '__main__': run([ "price_script_unity.py", "android", "/Users/zhuge/Temp/20201226/test/archery_inter_20200805_20200904_summary.txt", "/Users/zhuge/Temp/20201226/test/unity.csv", ])