← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to utilities/format-imports

  • Committer: Curtis Hovey
  • Date: 2011-12-18 15:13:07 UTC
  • mto: This revision was merged to the branch mainline in revision 14547.
  • Revision ID: curtis.hovey@canonical.com-20111218151307-sdm2gzobt5tplbe0
Moved badges to lp.app.

Show diffs side-by-side

added added

removed removed

Lines of Context:
utilities/format-imports
152
152
comment_regex = re.compile(
153
153
    "(?P<comment>(^#.+\n)+)(^import|^from) +(?P<module>[a-zA-Z0-9_.]+)", re.M)
154
154
split_regex = re.compile(",\s*")
155
 
module_base_regex = re.compile("([^. ]+)")
156
155
 
157
156
# Module docstrings are multiline (""") strings that are not indented and are
158
157
# followed at some point by an import .
286
285
 
287
286
    return imports
288
287
 
289
 
LOCAL_PACKAGES = (
290
 
    'canonical', 'lp', 'launchpad_loggerhead', 'devscripts',
291
 
    # database/* have some implicit relative imports.
292
 
    'fti', 'replication', 'preflight', 'security', 'upgrade',
293
 
    )
294
288
 
295
289
def format_imports(imports):
296
290
    """Group and order imports, return the new import statements."""
297
 
    early_section = {}
298
291
    standard_section = {}
299
292
    first_section = {}
300
293
    thirdparty_section = {}
301
294
    local_section = {}
302
295
    # Group modules into sections.
303
296
    for module, statement in imports.iteritems():
304
 
        module_base = module_base_regex.findall(module)[0]
 
297
        module_base = module.split('.')[0]
305
298
        comment = statement.comment
306
 
        if module_base == '_pythonpath':
307
 
            early_section[module] = statement
308
 
        elif comment is not None and comment.startswith("# FIRST"):
 
299
        if comment is not None and comment.startswith("# FIRST"):
309
300
            first_section[module] = statement
310
 
        elif module_base in LOCAL_PACKAGES:
 
301
        elif module_base in ('canonical', 'lp'):
311
302
            local_section[module] = statement
312
303
        elif module_base in python_standard_libs:
313
304
            standard_section[module] = statement
317
308
    all_import_lines = []
318
309
    # Sort within each section and generate statement strings.
319
310
    sections = (
320
 
        early_section,
321
311
        standard_section,
322
312
        first_section,
323
313
        thirdparty_section,
349
339
    imports_section = pyfile[import_start:import_end]
350
340
    imports = parse_import_statements(imports_section)
351
341
 
352
 
    next_char = pyfile[import_end:import_end + 1]
353
 
 
354
 
    if next_char == '':
355
 
        number_of_newlines = 1
356
 
    elif next_char != '#':
 
342
    if pyfile[import_end:import_end + 1] != '#':
357
343
        # Two newlines before anything but comments.
358
344
        number_of_newlines = 3
359
345
    else: