update: python functions

Author: ui2code

source/duplication.py

@@ -0,0 +1,9 @@
+#!/usr/bin/env python3
+
+def list_duplicates(seq):
+  seen = set()
+  seen_add = seen.add
+  # adds all elements it doesn't know yet to seen and all other to seen_twice
+  seen_twice = set( x for x in seq if x in seen or seen_add(x) )
+  # turn the set into a list (as requested)
+  return list( seen_twice )

source/json_generator.py

@@ -4,7 +4,7 @@
 import fnmatch
 from json import dumps
 
-def writeJson(data, filename):
+def write_json(data, filename):
   try:
     jsondata = dumps(data, indent=2, skipkeys=True, sort_keys=True)
     fd = open(filename, 'w')
@@ -14,16 +14,16 @@ def writeJson(data, filename):
     print 'ERROR writing', filename
     pass
 
-def getDir(path, ext):
+def get_dir(path, ext):
   matches = []
   for root, dirnames, filenames in os.walk(path):
     for filename in fnmatch.filter(filenames, ext):
       matches.append(os.path.join(root, filename))
   return matches
 
-def getJson():
+def get_json():
   result = []
-  for count, file in enumerate(getDir('../snippets/', '*.sublime-snippet'), start=1):
+  for count, file in enumerate(get_dir('../snippets/', '*.sublime-snippet'), start=1):
     line        = open(file)
     content     = line.read()
     trigger     = content.split('<tabTrigger>')[1].split('</tabTrigger>')[0]
@@ -34,5 +34,5 @@ def getJson():
   return result
 
 
-writeJson(getJson(), "../snippets.json")
+write_json(get_json(), "../snippets.json")
 

tests/test.py

@@ -2,23 +2,16 @@
 
 import sys
 sys.path.append('../source/')
-from json_generator import getJson
+from json_generator import get_json
+from duplication import list_duplicates
 import json
 import re
 import unittest
 
-def list_duplicates(seq):
-  seen = set()
-  seen_add = seen.add
-  # adds all elements it doesn't know yet to seen and all other to seen_twice
-  seen_twice = set( x for x in seq if x in seen or seen_add(x) )
-  # turn the set into a list (as requested)
-  return list( seen_twice )
-
 class TestDuplicatesMethods(unittest.TestCase):
 
   def test_duplicates(self):
-    json = getJson()
+    json = get_json()
     triggers = []
 
     for snippet in json: