Ejemplo n.º 1
0
    def remove_from_db(picture):
        with get_db().atomic() as txn:
            fragments_to_delete = [f for f in picture.fragments]
            for f in fragments_to_delete:
                f.delete_instance()

            picture.delete_instance()
Ejemplo n.º 2
0
    def remove_from_db(picture):
        with get_db().atomic() as txn:
            fragments_to_delete = [f for f in picture.fragments]
            for f in fragments_to_delete:
                f.delete_instance()

            picture.delete_instance()
Ejemplo n.º 3
0
    def save_to_db(picture, json_data):
        print 'START SAVING'
        data = json.loads(json_data)
        print 'CONVERTED'

        try:
            with get_db().atomic() as txn:
                picture.global_diff = data['globalDiff']
                picture.tag = data['tagName'] if 'tagName' in data else None
                picture.next_tag_id = data[
                    'next_max_tag_id'] if 'next_max_tag_id' in data else None
                picture.updated = datetime.datetime.now()

                picture.save()
                print 'PICTURE SAVED'

                print 'ALL FRAGMENTS SELECTED'

                for g in data['updatedGroups']:
                    upd = Fragment.update(x=g['x'], y=g['y'], diff=g['diff'])
                    upd = upd.where(Fragment.id == g['id'],
                                    Fragment.picture == picture)
                    upd.execute()

                print 'UPDATED: %d' % len(data['updatedGroups'])

                rc = Fragment.select()
                rc = rc.where(Fragment.id << data['removedPicrures'],
                              Fragment.picture == picture)
                rc = rc.count()

                if rc:
                    dlt = Fragment.delete()
                    dlt = dlt.where(Fragment.id << data['removedPicrures'],
                                    Fragment.picture == picture)
                    dlt.execute()

                print 'REMOVED: %d' % rc

                to_remove = [
                    f.id for f in picture.fragments.where(Fragment.x == None)
                ]
                if len(to_remove) > 0:
                    Fragment.delete().where(Fragment.id << to_remove).execute()
                print 'EXTRA FRAGMENTS REMOVED: %d' % len(to_remove)
            return True
        except WrongDataException:
            return False
Ejemplo n.º 4
0
    def save_to_db(picture, json_data):
        print 'START SAVING'
        data = json.loads(json_data)
        print 'CONVERTED'

        try:
            with get_db().atomic() as txn:
                picture.global_diff = data['globalDiff']
                picture.tag = data['tagName'] if 'tagName' in data else None
                picture.next_tag_id = data['next_max_tag_id'] if 'next_max_tag_id' in data else None
                picture.updated = datetime.datetime.now()

                picture.save()
                print 'PICTURE SAVED'

                print 'ALL FRAGMENTS SELECTED'

                for g in data['updatedGroups']:
                    upd = Fragment.update(x=g['x'], y=g['y'], diff=g['diff'])
                    upd = upd.where(Fragment.id == g['id'], Fragment.picture == picture)
                    upd.execute()

                print 'UPDATED: %d' % len(data['updatedGroups'])

                rc = Fragment.select()
                rc = rc.where(Fragment.id << data['removedPicrures'], Fragment.picture == picture)
                rc = rc.count()

                if rc:
                    dlt = Fragment.delete()
                    dlt = dlt.where(Fragment.id << data['removedPicrures'], Fragment.picture == picture)
                    dlt.execute()

                print 'REMOVED: %d' % rc

                to_remove = [f.id for f in picture.fragments.where(Fragment.x == None)]
                if len(to_remove) > 0:
                    Fragment.delete().where(Fragment.id << to_remove).execute()
                print 'EXTRA FRAGMENTS REMOVED: %d' % len(to_remove)
            return True
        except WrongDataException:
            return False