itertools学习


作为一个写业务逻辑的boy,我需要专注的就是把业务逻辑写好。写业务逻辑并不复杂,就是把编程最基础的东西使用好,有变量循环流程控制函数数据库等。
但是写出的逻辑要通俗易懂、易于理解,避免炫技、晦涩。在目前团队中待了一段时间了,根据 code review 和自己的理解整理出一些基本原则。

业务代码思想原则

第一:代码要足够容易阅读。

代码是给程序员阅读的,首先要做到的就是逻辑清晰,易于阅读。所以要避免毫无意义的变量名、嵌套过多的逻辑判断、过于复杂的流程控制。在易读性和性能两者之间,首先要保证易读性,当然两者并不冲突。

第二:代码性能要有保证

python代码的执行速度比不上编译性语言,所以要在代码层面上保证性能的可靠。通常解决一个主要降低性能的点,就能提升整体的性能。

业务代码编码原则

变量

核心思想:清晰表示变量的功能
原则

  1. 变量命名统一,python推荐使用下划线分隔命名法,即蛇形命名法,毕竟python就是蛇嘛。真实原因:Python 为什么要推荐蛇形命名法
  2. 变量命名中可以带数据类型,如 buy_user_listid_to_name_dict 。python的灵活性高,为了更加方便区分变量、函数、方法等,变量名加数据类型是个好方法
  3. 超过三个的常量都要用枚举值。善用枚举值,让枚举值的名字解释常量,这是遵守第一条原则,容易阅读
  4. 多使用局部变量少使用全局变量,命名空间中局部变量优先搜索

流程控制

核心思想:减少if else嵌套,逻辑越清晰越好
原则
减少if else

  1. 可以使用字典的key value特性,直接用key命中条件,避免if判断。
  2. 使用装饰器装饰函数,装饰器实现里用字典特性做if else判断。可参考:如何简化大量的 if…elif…else 代码?
  3. 用in操作替换if else判断。
  4. 使用any 或 all 将多个判断一起处理,减少if else的分支

无法减少if else时要让逻辑清晰

  1. 存在复杂的判断条件时,可以先将判断条件处理一遍。如一个判断(a and b) or (c and d)操作时,先单独判断出 a and b 和 c and d,然后再判断两者结果
  2. 减少if else之间的处理流程。if或else中处理的流程过长时要尽量简化流程处理,抽象出函数。让if else对齐,处理的逻辑看起来清晰。
  3. if条件的短路特性。if a or b 这种判断中,如果a是True就不会判断b,所以将True条件写在前面可以节省判断时间

简单小例子:使用 all 减少 if else


have_opencourse = True if plan_id in [1,2] else False

user_info = bayuser.get_user_info(user_id)
register_time = pendulum.parse(user_info.created_at)
register_at_today = True if register_time.date() == get_local_today() else False

checkin_days = uc.get_user_checkin_days(user_id)
checkin_zero_day = True if checkin_days == 0 else False

ab_test_id = 414
ab_test_name = words_utils.get_user_plan(user_id, ab_test_id).user_plan
no_provide = True if ab_test_name == "no_provide" else False

# 使用if else判断
if register_time.date() == get_local_today() and checkin_days == 0 and ab_test_name == "no_provide" and plan_id in [1,2]:
    return encoder.json_response(code=200)
    
# 使用all判断
if all([have_opencourse, register_at_today, checkin_zero_day, no_provide]):
    return encoder.json_response(code=200)

循环

核心思想:减少for循环
原则

  1. 用for循环代替while循环,for循环比while循环快
  2. 使用隐式for循环代替显式for循环。如sum,map,filter,reduce等都是隐式for循环
  3. 尽量不要打断循环。打断循环的放在外面。有判断条件的语句和与循环不相关的操作语句尽量放在for外面
  4. 应当将最长的循环放在最内层,最短的循环放在最外层,以减少CPU跨切循环层的次数
  5. for循环和集合都可以处理的选择集合解决,集合的效率远高于循环

简单小例子:集合替代for循环

# for循环
for user in pay_user_cet4_set:
    if user not in all_presale_user_set:
        cet4_push.append(user)

# 集合操作
cet4_push = pay_user_cet4_set - all_presale_user_set

函数

核心思想:功能单一,短小精悍
原则

  1. 函数设计要尽量短小,嵌套层次不宜过深
  2. 一个函数只做一件事,尽量保证函数语句粒度的一致性。
  3. 函数申明应该做到合理,简单,易于使用。函数名合理,参数不宜过多。
  4. 函数参数设计应该考虑向下兼容。通过添加默认参数来实现。

小结

之前分享过一本如何优雅写python的书,可以看之前写的一篇文章:

在每一个开发团队都有会团队的代码风格。优秀的代码风格让团队中庞大的代码井井有条,处处有规律可循,就像是一个人完成的。每一个人也要有自己的代码原则,缺少基本原则的指导会让代码看起来不和谐,前后矛盾,这样肯定是不利于个人成长的。

本文提到的原则并见得全部正确,只是我个人在实际工作中的真实总结,绝不是闭门造车,仅供参考,欢迎交流。后续总结的原则也会继续更新到这里。