Rehabilitate and/or deter. The extent to which it is either the former or the latter depends on the country. That also determines whether or not "treating them badly" should be a feature of the justice system.