The problem with your argument is that even if you understand your child and know that smartphone use in school is distracting them from education, you can't really ban them from using a smartphone of everyone else is allowed, or you risk them becoming a social pariah.

School-wide and broader bans actually fix this problem of social needs, by leveling the playing field.