as_partition_key for belongs_to