哺乳期内，公司辞退是否属于违法行为？可以怎么维权？

因为公司有条制度是员工不可以谈恋爱，刚开始不知道这个规定，进入公司后和老公认识到后来结婚，我们部门领导说如果我们结婚必定要辞退一个，当时我已经有宝贝了，孕妇他们不敢辞退，经理给老总说生完孩子就辞职，现在月子还没有完就找我老公谈话了，想想心里很不舒服，怀孕期间尿频上厕所说我到处走动，公司停电让我扛着主机去网吧做资料，生宝贝的前一天还在上班，休息还只能轮休。想到受了那么多委屈，就这样叫我辞职，心里肯定不舒服，该替自己争取一下个人利益！有没有学法律方面的宝妈，可以支点招吗？