Merge pull request joke2k#277 from ericchaves/master

added support for brazilian SSN (called CPF)

Author: joke2k

faker/providers/ssn/pt_BR/__init__.py

@@ -0,0 +1,43 @@
+#  -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+from .. import Provider as SsnProvider
+from faker.generator import random
+
+
+def checksum(digits):
+    s = 0
+    p = len(digits) + 1
+    for i in range(0, len(digits)):
+        s += digits[i] * p
+        p -= 1
+
+    reminder = s % 11
+    if reminder == 0 or reminder == 1:
+        return 1
+    else:
+        return 11 - reminder
+
+
+class Provider(SsnProvider):
+    """
+    Provider for Brazilian SSN also known in Brazil as CPF.
+    There are two methods Provider.ssn and Provider.cpf
+    The snn returns a valid number with numbers only
+    The cpf return a valid number formatted with brazilian mask. eg nnn.nnn.nnn-nn
+    """
+
+    @classmethod
+    def ssn(cls):
+        digits = random.sample(range(10), 9)
+
+        dv = checksum(digits)
+        digits.append(dv)
+        digits.append(checksum(digits))
+
+        return ''.join(map(str, digits))
+
+    @classmethod
+    def cpf(cls):
+        c = Provider.ssn()
+        return c[:3] + '.' + c[3:6] + '.' + c[6:9] + '-' + c[9:]

faker/tests/pt_BR/__init__.py

@@ -0,0 +1,26 @@
+#  -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+import unittest
+import re
+
+from faker import Factory
+from faker.providers.ssn.pt_BR import Provider, checksum
+
+
+class pt_BR_FactoryTestCase(unittest.TestCase):
+    def setUp(self):
+        self.factory = Factory.create('pt_BR')
+
+    def test_pt_BR_ssn_checksum(self):
+        self.assertEqual(checksum([8, 8, 2, 8, 2, 1, 6, 5, 2]), 2)
+        self.assertEqual(checksum([8, 8, 2, 8, 2, 1, 6, 5, 2, 2]), 1)
+
+    def test_pt_BR_ssn(self):
+        for i in range(100):
+            self.assertTrue(re.search(r'^\d{11}$', Provider.ssn()))
+
+    def test_pt_BR_cpf(self):
+        for i in range(100):
+            self.assertTrue(re.search(r'\d{3}\.\d{3}\.\d{3}\-\d{2}', Provider.cpf()))