Esempio n. 1
0
File: git-db.py Progetto: cb22/gitdb
def status():
    branch = git.current_branch()
    print "# On branch %s, tracking database %s" % (branch, database)
    
    tracked = [db[len(database):] for db in list_databases() if db.startswith(database)]
    tracked[tracked.index("")] = branch
    print "# Currently tracking local branches: %s" % tracked

    print "# Patches applied:"
    print "# %s" % list_applied_patches(branch, database)
    print "# Patches not applied:"
    print "# %s" % [patch for patch in list_file_patches(branch, database) if patch not in list_applied_patches(branch, database) and patch not in list_ignored_patches(branch, database)]
    print "# Patches ignored:"
    print "# %s" % list_ignored_patches(branch, database)
    print "# Patches stashed:"
    print "# %s" % list_stashed_patches(branch, database)


    last_commit = git.last_commit_time()
    if fast_check_sqlchanges(database, last_commit):
        with amalgamated_sql(branch, database) as amalgamated_sql_file:
            sql_difference = calculate_difference(amalgamated_sql_file, database, [filters.filter_auto_increment, lambda input: filters.filter_renames(input, database, git.last_commit_time())])
        if sql_difference:
            print "# Untracked SQL code since last commit at '%s'." % (last_commit)
        else:
            print "# SQL database clean."
    else:
        print "# SQL database clean."
Esempio n. 2
0
File: git-db.py Progetto: cb22/gitdb
def stash():
    branch = git.current_branch()
    if args.unstash or args.delete:
        if list_stashed_patches(branch, database):
            patches = dict([(i, patch) for i, patch in enumerate(list_stashed_patches(branch, database))])
            pprint(patches)
            try:
                input = raw_input("Please enter a space separated list of patch #s to unstash: " if args.unstash else "Please enter a space separated list of patch #s to delete: ")
                if input != "":
                    to_unstash = [patches[int(i)] for i in input.split(" ")]
                    
                    for unstashed in to_unstash:
                        if args.unstash:
                            try:
                                apply_patch(database, unstashed, True)
                                remove_stashed_from_db(branch, database, unstashed)
                            except MySqlException:
                                print "Error applying stashed patch %s, not continuing"
                        elif args.delete:
                            # Delete the patch here.
                            os.remove(unstashed)
                            remove_stashed_from_db(branch, database, unstashed)
                        
            except:
                print "Invalid input."
                return 
            
        else:
            print "No stashed patches available for unstashing." if args.unstash else "No stashed patches available for deleting." 
            
        if args.unstash:
            add_event_to_db(database, "unstashed patches, %s: ", time.time())
    else:
        if fast_check_sqlchanges(database, git.last_commit_time()):
            print "Stashing all schema changes made on branch %s to database %s" % (branch, database)
            
            with amalgamated_sql(branch, database) as amalgamated_sql_file:
                base_dir = os.path.join(".git", "gitdb", "stashed", branch, database)
                patch_name = os.path.join(base_dir, free_patch_name())
                makedirs(base_dir)
                
                sql_changes = calculate_difference(amalgamated_sql_file, database, [filters.filter_auto_increment, lambda input: filters.filter_renames(input, database, git.last_commit_time())])
                if not sql_changes:
                    print "No schema changes to stash."
                    return
                else:
                    print "Your SQL schema changes will be stashed in the file %s. You can use the git db stash command to manipulate them." % patch_name
                    with open(patch_name, "w") as sql_patch:
                        sql_patch.write(sql_changes)
        
                    # Rollback the database to the last commit.
                    temp_file = NamedTemporaryFile()
                    sql_difference = calculate_difference(database, amalgamated_sql_file, [filters.filter_auto_increment, lambda input: filters.filter_renames(input, database, git.last_commit_time())])
                    temp_file.write(sql_difference)
                    temp_file.flush()
                    
                    apply_patch(database, temp_file.name, True)
                    temp_file.close()
                    
                    add_stashed_to_db(branch, database, patch_name)
                    add_event_to_db(database, "stash", time.time())
        else:
            print "No schema changes to stash."