公司要求因上班玩手机而返工或采取其他纪律处分措施是否合法，主要取决于公司的规章制度是否明确、合理且经过法定程序制定与公示，以及员工玩手机行为的严重程度和对工作的影响。以下是对这一问题的详细分析：
一、公司规章制度的规定
明确禁止条款：
- 如果公司的规章制度中明确规定了员工在工作期间不得使用手机或进行与工作无关的活动，并且这一规定已经通过合法程序（如职工代表大会讨论、公示等）告知了员工，那么员工违反此规定可能构成严重违纪。在此情况下，公司有权依据规章制度要求员工返工或采取其他纪律处分措施。
处罚措施合理：
- 若规章制度中还规定了对于此类违纪行为的相应处罚措施，包括但不限于警告、罚款、降职、解除劳动合同等，且该处罚措施与违纪行为的严重程度相适应，那么公司在员工多次违反规定或情节严重时，有权依据规章制度采取相应的纪律处分措施。
二、实际情况的考量
玩手机的性质：
- 并非所有在工作时间玩手机的行为都构成严重违纪。例如，短暂的查看信息、接听紧急电话等，可能不会对工作造成实质性影响。然而，长时间玩游戏、刷视频等行为则明显属于与工作无关的活动，可能严重影响工作效率和团队氛围。对于此类行为，公司有权依据规章制度采取纪律处分措施。
对工作的影响：
- 公司需要评估员工玩手机行为对工作的影响程度。如果因为玩手机导致工作任务未完成、工作质量下降或给公司带来损失，那么公司有权采取相应的纪律处分措施，包括要求员工返工以弥补损失。
证据收集：
- 在决定对员工采取纪律处分措施之前，公司应确保有充分的证据证明员工存在玩手机且违反规章制度的行为。这包括监控录像、同事证言、工作记录等。证据的充分性和合法性是公司采取纪律处分措施的重要依据。
三、法律程序的遵循
通知工会：
- 根据《劳动合同法》第四十三条的规定，用人单位单方解除劳动合同应当事先将理由通知工会。虽然这一条款主要针对的是解除劳动合同的情形，但公司在采取其他纪律处分措施时，也应遵循类似的程序，确保程序的合法性和公正性。
出具书面证明：
- 如果公司决定对员工采取纪律处分措施，如要求返工、罚款等，应向员工出具书面的处分决定，并说明处分的原因和依据。这有助于保障员工的知情权，也有助于避免后续的法律纠纷。
四、特殊情况处理
规章制度缺失或未公示：
- 若公司没有相关明确规定，或虽有规定但未履行法定公示或告知程序，直接要求员工返工或采取其他纪律处分措施可能构成违法。在此情况下，员工有权拒绝公司的要求，并通过法律途径维护自己的合法权益。
玩手机情节轻微：
- 对于员工偶尔、短暂地玩手机且未对工作造成实质性影响的行为，公司直接要求返工或采取其他严厉的纪律处分措施可能属于滥用管理权。在此情况下，员工同样可以通过法律途径维护自己的合法权益。
五、建议与指导
对公司：
- 建议公司制定完善、合理的规章制度，并确保其经过民主程序制定与公示。在处理员工玩手机问题时，应严格遵循法律法规和公司规章制度，确保处分决定的合法性和合理性。同时，公司也应注重与员工的沟通和教育，引导员工自觉遵守规章制度，提高工作效率。
对员工：
- 建议员工了解并遵守公司的规章制度，避免在工作时间过度使用手机而影响工作。如遇公司不当要求或处分决定，应及时向我咨询并通过法律途径维护自己的权益。
综上所述，公司要求因上班玩手机而返工或采取其他纪律处分措施是否合法是一个复杂的问题，需要综合考虑多个因素。在实际操作中，公司和员工都应遵循相关法律法规和规章制度，确保各自行为的合法性和合理性。如有任何疑问或需要进一步的法律帮助，请直接向我咨询。