Countries that refuse to accept returns of individuals in the UK unlawfully could see the number of visas they are granted cut.